Skip to content

Commit 606ebda

Browse files
committed
cache to chemlog extra
1 parent c304092 commit 606ebda

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

chebifier/prediction_models/chemlog_predictor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ def __init__(self, model_name: str, **kwargs):
5555
self.chebi_graph = kwargs.get("chebi_graph", None)
5656
self.classifier = self.CHEMLOG_CLASSIFIER()
5757

58+
@modelwise_smiles_lru_cache.batch_decorator
5859
def predict_smiles_list(self, smiles_list: list[str]) -> list:
5960
mol_list = [_smiles_to_mol(smiles) for smiles in smiles_list]
6061
res = self.classifier.classify(mol_list)

0 commit comments

Comments
 (0)