Skip to content

Commit

Permalink
fix: continue_on_num_success/continue_on_success_ratio key error
Browse files Browse the repository at this point in the history
wangzyphysics committed Jun 3, 2024

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
1 parent fa4c0db commit efbe63d
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions dpgen2/superop/caly_evo_step.py
Original file line number Diff line number Diff line change
@@ -147,6 +147,9 @@ def _caly_evo_step(
run_executor_config = run_config.pop("executor")
template_slice_config = run_config.pop("template_slice_config", {})
expl_mode = caly_evo_step_steps.expl_mode
no_slice_run_config = deepcopy(run_config)
no_slice_run_config.pop("continue_on_num_success", None)
no_slice_run_config.pop("continue_on_success_ratio", None)

def wise_executor(expl_mode, origin_executor_config):
if expl_mode == "default":
@@ -185,7 +188,7 @@ def wise_executor(expl_mode, origin_executor_config):
caly_evo_step_steps.inputs.parameters["cnt_num"],
),
executor=wise_executor(expl_mode, prep_executor_config),
**run_config,
**no_slice_run_config,
)
caly_evo_step_steps.add(collect_run_calypso)

@@ -219,7 +222,7 @@ def wise_executor(expl_mode, origin_executor_config):
caly_evo_step_steps.inputs.parameters["cnt_num"],
),
executor=wise_executor(expl_mode, prep_executor_config),
**run_config,
**no_slice_run_config,
)
caly_evo_step_steps.add(prep_dp_optim)

0 comments on commit efbe63d

Please sign in to comment.