chore: rename expl_config into config in prepcalymodeldevi op

wangzyphysics · Jun 1, 2024 · 0c1d8ab · 0c1d8ab
1 parent e0b923f
commit 0c1d8ab
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/dpgen2/op/prep_caly_model_devi.py b/dpgen2/op/prep_caly_model_devi.py
@@ -48,7 +48,7 @@ def get_input_sign(cls):
         return OPIOSign(
             {
                 "task_name": Parameter(str),
-                "expl_config": BigParameter(dict),
+                "config": BigParameter(dict),
                 "traj_results": Artifact(List[Path]),
             }
         )
@@ -74,7 +74,7 @@ def execute(
         ip : dict
             Input dict with components:
             - `task_name` : (`str`)
-            - `expl_config` : (`BigParameter(dict)`)
+            - `config` : (`BigParameter(dict)`)
             - `traj_results` : (`Path`)
 
         Returns
@@ -93,7 +93,7 @@ def execute(
             for traj_dir in traj_results_dir
             for traj in Path(traj_dir).rglob("*.traj")
         ]
-        expl_config = ip["expl_config"]
+        expl_config = ip["config"]
         group_size = expl_config.get("model_devi_group_size", len(trajs))
 
         with set_directory(work_dir):

diff --git a/dpgen2/superop/prep_run_calypso.py b/dpgen2/superop/prep_run_calypso.py
@@ -253,7 +253,7 @@ def _prep_run_caly(
         ),
         parameters={
             "task_name": "prep-calypso-model-deviation",
-            "expl_config": prep_run_caly_steps.inputs.parameters["explore_config"],
+            "config": prep_run_caly_steps.inputs.parameters["explore_config"],
         },
         artifacts={
             "traj_results": caly_evo_step.outputs.artifacts["traj_results"],

diff --git a/tests/op/test_prep_caly_model_devi.py b/tests/op/test_prep_caly_model_devi.py
@@ -72,7 +72,7 @@ def test_00_success(self):
             OPIO(
                 {
                     "task_name": self.run_dir_name,
-                    "expl_config": explore_config,
+                    "config": explore_config,
                     "traj_results": self.ref_traj_results,
                 }
             )
@@ -100,7 +100,7 @@ def test_01_success(self):
             OPIO(
                 {
                     "task_name": self.run_dir_name,
-                    "expl_config": explore_config,
+                    "config": explore_config,
                     "traj_results": self.ref_traj_results,
                 }
             )