Update cgcnn_inference.py

globus-labs · Apr 10, 2024 · a044b66 · a044b66
1 parent 1c04a18
commit a044b66
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/mofa/scoring/cgcnn_inference.py b/mofa/scoring/cgcnn_inference.py
@@ -1,7 +1,6 @@
 import os
 import ase
 from pymatgen.io.ase import AseAtomsAdaptor
-from train.dist_utils import get_local_rank, init_distributed
 import torch.nn as nn
 import torch.distributed as dist
 from torch_geometric.loader import DataLoader