Update difflinker_sample_and_analyze.py

globus-labs · Nov 9, 2023 · 3e0d95b · 3e0d95b
1 parent 956e438
commit 3e0d95b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/mofa/utils/difflinker_sample_and_analyze.py b/mofa/utils/difflinker_sample_and_analyze.py
@@ -7,11 +7,11 @@
 import torch
 from rdkit import Chem
 
-from utils.src import const
-from utils.src.datasets import collate_with_fragment_edges, get_dataloader, parse_molecule
-from utils.src.lightning import DDPM
-from utils.src.linker_size_lightning import SizeClassifier
-from utils.src.visualizer import save_xyz_file, visualize_chain
+from mofa.utils.src import const
+from mofa.utils.src.datasets import collate_with_fragment_edges, get_dataloader, parse_molecule
+from mofa.utils.src.lightning import DDPM
+from mofa.utils.src.linker_size_lightning import SizeClassifier
+from mofa.utils.src.visualizer import save_xyz_file, visualize_chain
 from tqdm import tqdm