use chebi-utils

Author: sfluegel05

chebai_graph/preprocessing/reader/augmented_reader.py

@@ -4,7 +4,7 @@
 
 import torch
 from chebai.preprocessing.reader import DataReader
-from chebai.preprocessing.datasets.chebi import sanitize_molecule
+from chebi_utils.sdf_extractor import sanitize_molecule
 from rdkit import Chem
 from torch_geometric.data import Data as GeomData