Fix imports.

mila-iqia · Oct 22, 2024 · 56bb596 · 56bb596
1 parent 9dbf6a0
commit 56bb596
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/data/crop_lammps_outputs.py b/data/crop_lammps_outputs.py
@@ -5,8 +5,10 @@
 
 import yaml
 
-from crystal_diffusion.data.utils import crop_lammps_yaml
-from crystal_diffusion.utils.logging_utils import setup_analysis_logger
+from diffusion_for_multi_scale_molecular_dynamics.data.utils import \
+    crop_lammps_yaml
+from diffusion_for_multi_scale_molecular_dynamics.utils.logging_utils import \
+    setup_analysis_logger
 
 logger = logging.getLogger(__name__)