Update cgcnn_inference.py

globus-labs · Apr 10, 2024 · cbb42fa · cbb42fa
1 parent 250ba2b
commit cbb42fa
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mofa/scoring/cgcnn_inference.py b/mofa/scoring/cgcnn_inference.py
@@ -1,5 +1,3 @@
-import os
-import ase
 from pymatgen.io.ase import AseAtomsAdaptor
 import torch.nn as nn
 import torch.distributed as dist
@@ -9,7 +7,9 @@
 from torch.nn.parallel import DistributedDataParallel
 from typing import Union
 from pathlib import Path
-from models import cgcnn, CrystalGraphConvNet
+from mofa.models.cgcnn import CrystalGraphConvNet
+import os
+import ase
 import abc
 import json
 import torch