diff --git a/epdb/models.py b/epdb/models.py index 87bac4af..5f5471a6 100644 --- a/epdb/models.py +++ b/epdb/models.py @@ -3471,9 +3471,7 @@ class EnviFormer(PackageBasedModel): def predict_batch(self, smiles: List[str], *args, **kwargs): # Standardizer removes all but one compound from a raw SMILES string, so they need to be processed separately canon_smiles = [ - ".".join( - [FormatConverter.standardize(s, remove_stereo=True) for s in smiles.split(".")] - ) + ".".join([FormatConverter.standardize(s, remove_stereo=True) for s in smi.split(".")]) for smi in smiles ] logger.info(f"Submitting {canon_smiles} to {self.get_name()}")