fix imports in scripts

FAIR-Chem · Jul 22, 2024 · f7bae74 · f7bae74
1 parent c2896a5
commit f7bae74
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/fairchem/core/scripts/fit_normalizers.py b/src/fairchem/core/scripts/fit_normalizers.py
@@ -13,8 +13,10 @@
 
 from fairchem.core.common.registry import registry
 from fairchem.core.common.utils import load_config, save_checkpoint
-from fairchem.core.modules.element_references import create_element_references
-from fairchem.core.modules.normalizer import fit_normalizers
+from fairchem.core.modules.normalization.element_references import (
+    create_element_references,
+)
+from fairchem.core.modules.normalization.normalizer import fit_normalizers
 
 
 def fit_norms(

diff --git a/src/fairchem/core/scripts/fit_references.py b/src/fairchem/core/scripts/fit_references.py
@@ -13,7 +13,7 @@
 
 from fairchem.core.common.registry import registry
 from fairchem.core.common.utils import load_config, save_checkpoint
-from fairchem.core.modules.element_references import fit_linear_references
+from fairchem.core.modules.normalization.element_references import fit_linear_references
 
 
 def fit_linref(config: dict, output_path: str | Path) -> None: