dyanmic import for nx

Author: aditya0by0

chebai_graph/preprocessing/reader.py

@@ -2,7 +2,6 @@
 from typing import List, Optional
 
 import chebai.preprocessing.reader as dr
-import networkx as nx
 import pysmiles as ps
 import rdkit.Chem as Chem
 import torch
@@ -93,6 +92,8 @@ def name(cls):
         return "graph"
 
     def _read_data(self, raw_data) -> Optional[GeomData]:
+        import networkx as nx
+
         # raw_data is a SMILES string
         try:
             mol = ps.read_smiles(raw_data)