Update difflinker_train.py

globus-labs · Nov 9, 2023 · 2a9f17e · 2a9f17e
1 parent 3e0d95b
commit 2a9f17e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mofa/difflinker_train.py b/mofa/difflinker_train.py
@@ -7,9 +7,9 @@
 from datetime import datetime
 from pytorch_lightning import Trainer, callbacks, loggers
 
-from utils.src.const import NUMBER_OF_ATOM_TYPES, GEOM_NUMBER_OF_ATOM_TYPES
-from utils.src.lightning import DDPM
-from utils.src.utils import disable_rdkit_logging, Logger
+from mofa.utils.src.const import NUMBER_OF_ATOM_TYPES, GEOM_NUMBER_OF_ATOM_TYPES
+from mofa.utils.src.lightning import DDPM
+from mofa.utils.src.utils import disable_rdkit_logging, Logger
 
 from pytorch_lightning.callbacks import TQDMProgressBar