Skip to content

Commit 65cd5f1

Browse files
Fix parameter check
1 parent d8db0b0 commit 65cd5f1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

janus_core/helpers/utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -257,7 +257,7 @@ def results_to_info(
257257
if struct.calc and "arch" in struct.calc.parameters:
258258
arch = struct.calc.parameters["arch"]
259259
struct.info["arch"] = arch
260-
if struct.calc and "mlip_model" in struct.calc.parameters:
260+
if struct.calc and "model" in struct.calc.parameters:
261261
struct.info["mlip_model"] = struct.calc.parameters["model"]
262262

263263
for key in properties & struct.calc.results.keys():
@@ -476,7 +476,7 @@ def output_structs(
476476
for image in images:
477477
if image.calc and "arch" in image.calc.parameters:
478478
image.info["arch"] = image.calc.parameters["arch"]
479-
if image.calc and "mlip_model" in image.calc.parameters:
479+
if image.calc and "model" in image.calc.parameters:
480480
image.info["mlip_model"] = image.calc.parameters["model"]
481481

482482
# Add label for system

0 commit comments

Comments
 (0)