We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent be7278d commit a50ce8fCopy full SHA for a50ce8f
chebai_graph/preprocessing/datasets/chebi.py
@@ -130,7 +130,7 @@ def enc_if_not_none(encode, value):
130
131
# augment molecule graph if possible (this would also happen for the properties if needed, but this avoids redundancy)
132
if isinstance(self.reader, _AugmentorReader):
133
- returned_results = [self._create_augmented_graph(mol) for mol in features]
+ returned_results = [self.reader._create_augmented_graph(mol) for mol in features]
134
mols = [augmented_mol[1] for augmented_mol in returned_results if augmented_mol is not None]
135
else:
136
mols = features
0 commit comments