chore: restore parameter timeout in loop (#1022)

* restore parameter timeout in loop
This commit is contained in:
Tim
2025-07-05 14:55:26 +08:00
committed by GitHub
parent 6dbef13837
commit 4bf16d03cf
2 changed files with 3 additions and 3 deletions
+2 -2
View File
@@ -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__":
+1 -1
View File
@@ -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: