Skip to content

Commit a50ce8f

Browse files
committed
fix function call
1 parent be7278d commit a50ce8f

File tree

1 file changed

+1
-1
lines changed
  • chebai_graph/preprocessing/datasets

1 file changed

+1
-1
lines changed

chebai_graph/preprocessing/datasets/chebi.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def enc_if_not_none(encode, value):
130130

131131
# augment molecule graph if possible (this would also happen for the properties if needed, but this avoids redundancy)
132132
if isinstance(self.reader, _AugmentorReader):
133-
returned_results = [self._create_augmented_graph(mol) for mol in features]
133+
returned_results = [self.reader._create_augmented_graph(mol) for mol in features]
134134
mols = [augmented_mol[1] for augmented_mol in returned_results if augmented_mol is not None]
135135
else:
136136
mols = features

0 commit comments

Comments
 (0)