fix imports error

naik-aakash · Jan 10, 2025 · 2d79c34 · 2d79c34
1 parent c8f2f8e
commit 2d79c34
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/src/autoplex/auto/phonons/flows.py b/src/autoplex/auto/phonons/flows.py
@@ -19,6 +19,7 @@
     MPStaticSet,
 )
 
+from autoplex import MLIP_HYPERS
 from autoplex.auto.phonons.jobs import (
     complete_benchmark,
     dft_phonopy_gen_data,
@@ -33,10 +34,7 @@
 from autoplex.data.phonons.flows import IsoAtomStaticMaker, TightDFTStaticMaker
 from autoplex.data.phonons.jobs import reduce_supercell_size_job
 from autoplex.fitting.common.flows import MLIPFitMaker
-from autoplex.fitting.common.utils import (
-    MLIP_PHONON_DEFAULTS_FILE_PATH,
-    load_mlip_hyperparameter_defaults,
-)
+from autoplex.fitting.common.utils import MLIP_PHONON_DEFAULTS_FILE_PATH
 
 __all__ = [
     "CompleteDFTvsMLBenchmarkWorkflow",
@@ -279,9 +277,10 @@ def make(
         fit_input = {}
         bm_outputs = []
 
-        default_hyperparameters = load_mlip_hyperparameter_defaults(
-            mlip_fit_parameter_file_path=self.path_to_hyperparameters
-        )
+        # default_hyperparameters = load_mlip_hyperparameter_defaults(
+        #    mlip_fit_parameter_file_path=self.path_to_hyperparameters
+        # )
+        default_hyperparameters = MLIP_HYPERS.model_dump(by_alias=True)
 
         soap_default_dict = next(
             (