Skip to content

Commit 7fe4b83

Browse files
committed
optionally continue on error
1 parent 7fcc128 commit 7fe4b83

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

isaaclab_arena/evaluation/eval_runner.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,10 +138,11 @@ def main():
138138
metrics_logger.append_job_metrics(job.name, metrics)
139139

140140
except Exception as e:
141-
# continue with the next job even if one fails
142141
job_manager.complete_job(job, metrics={}, status=Status.FAILED)
143142
print(f"Job {job.name} failed with error: {e}")
144143
print(f"Traceback: {traceback.format_exc()}")
144+
if not args_cli.continue_on_error:
145+
raise
145146

146147
finally:
147148
# Only stop env if it was successfully created

isaaclab_arena/evaluation/eval_runner_cli.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,3 +14,9 @@ def add_eval_runner_arguments(parser: argparse.ArgumentParser) -> None:
1414
default="isaaclab_arena_environments/eval_jobs_configs/zero_action_jobs_config.json",
1515
help="Path to the eval jobs config file.",
1616
)
17+
parser.add_argument(
18+
"--continue_on_error",
19+
action="store_true",
20+
default=False,
21+
help="Continue evaluation with remaining jobs when a job fails instead of stopping immediately.",
22+
)

0 commit comments

Comments
 (0)