From 4bf16d03cf56af4c52e422173b6ec5353adf46cc Mon Sep 17 00:00:00 2001 From: Tim Date: Sat, 5 Jul 2025 14:55:26 +0800 Subject: [PATCH] chore: restore parameter timeout in loop (#1022) * restore parameter timeout in loop --- rdagent/app/data_science/loop.py | 4 ++-- rdagent/log/server/app.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/rdagent/app/data_science/loop.py b/rdagent/app/data_science/loop.py index d64c2e2f..c99c198a 100644 --- a/rdagent/app/data_science/loop.py +++ b/rdagent/app/data_science/loop.py @@ -14,7 +14,7 @@ def main( checkout: bool | str | Path = True, step_n: int | None = None, loop_n: int | None = None, - all_duration: str | None = None, + timeout: str | None = None, competition="bms-molecular-translation", replace_timer=True, exp_gen_cls: str | None = None, @@ -67,7 +67,7 @@ def main( if exp_gen_cls is not None: kaggle_loop.exp_gen = import_class(exp_gen_cls)(kaggle_loop.exp_gen.scen) - asyncio.run(kaggle_loop.run(step_n=step_n, loop_n=loop_n, all_duration=all_duration)) + asyncio.run(kaggle_loop.run(step_n=step_n, loop_n=loop_n, all_duration=timeout)) if __name__ == "__main__": diff --git a/rdagent/log/server/app.py b/rdagent/log/server/app.py index 807cd352..cca4e9fd 100644 --- a/rdagent/log/server/app.py +++ b/rdagent/log/server/app.py @@ -145,7 +145,7 @@ def upload_file(): if loop_n: cmds += ["--loop_n", loop_n] if all_duration: - cmds += ["--all_duration", f"{all_duration}h"] + cmds += ["--timeout", f"{all_duration}h"] app.logger.info(f"Started process for {log_trace_path} with parameters: {cmds}") with stdout_path.open("w") as log_file: