Fix parameter check

stfc · Oct 22, 2024 · 65cd5f1 · 65cd5f1
1 parent d8db0b0
commit 65cd5f1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/janus_core/helpers/utils.py b/janus_core/helpers/utils.py
@@ -257,7 +257,7 @@ def results_to_info(
     if struct.calc and "arch" in struct.calc.parameters:
         arch = struct.calc.parameters["arch"]
         struct.info["arch"] = arch
-    if struct.calc and "mlip_model" in struct.calc.parameters:
+    if struct.calc and "model" in struct.calc.parameters:
         struct.info["mlip_model"] = struct.calc.parameters["model"]
 
         for key in properties & struct.calc.results.keys():
@@ -476,7 +476,7 @@ def output_structs(
         for image in images:
             if image.calc and "arch" in image.calc.parameters:
                 image.info["arch"] = image.calc.parameters["arch"]
-            if image.calc and "mlip_model" in image.calc.parameters:
+            if image.calc and "model" in image.calc.parameters:
                 image.info["mlip_model"] = image.calc.parameters["model"]
 
     # Add label for system