Commit 3068f62e authored by nd-02110114's avatar nd-02110114
Browse files

🐛 fix path bug

parent 12340090

deepchem/models/__init__.py

+1 −1
Original line number Diff line number Diff line
@@ -30,7 +30,7 @@ from deepchem.models.chemnet_models import Smiles2Vec, ChemCeption 
# PyTorch models

try:

  from deepchem.models.torch_models import TorchModel

  from deepchem.models.torch_models import CGCNN

  from deepchem.models.torch_models import CGCNN, cgcnn_collate_fn

except ModuleNotFoundError:

  pass

deepchem/models/tests/test_cgcnn.py

+1 −2
Original line number Diff line number Diff line
@@ -3,9 +3,8 @@ from os import path, remove 


from deepchem.feat import CGCNNFeaturizer

from deepchem.molnet import load_perovskite

from deepchem.models import TorchModel, CGCNN, losses

from deepchem.metrics import Metric, mae_score

from deepchem.models.cgcnn import cgcnn_collate_fn

from deepchem.models import TorchModel, CGCNN, cgcnn_collate_fn, losses



try:

  import dgl  # noqa

deepchem/models/torch_models/__init__.py

+1 −1
Original line number Diff line number Diff line 
# flake8:noqa

from deepchem.models.torch_models.torch_model import TorchModel

from deepchem.models.torch_models.cgcnn import CGCNN
 
from deepchem.models.torch_models.cgcnn import CGCNN, cgcnn_collate_fn

CRA Git | Maintained and supported by SUSTech CRA and CCSE