results.append(
pool.apply_async(ComplexFeaturizer._featurize_callback,
(self, mol_file, protein_pdb, log_message)))
pool.close()
features = []
failures = []
for ind, result in enumerate(results):
new_features = result.get()
After Change
if not isinstance(complexes, Iterable):
complexes = [cast(Tuple[str, str], complexes)]
features = []
for i, point in enumerate(complexes):
if i % log_every_n == 0:
logger.info("Featurizing datapoint %i" % i)