diff --git a/lib_run_single.py b/lib_run_single.py index ab9816e..bcf2496 100644 --- a/lib_run_single.py +++ b/lib_run_single.py @@ -14,7 +14,7 @@ with open("./settings.json", "r") as file: data = json.load(file) time_limit = data["time_limit"] -# @timeout(time_limit, use_signals=False) +@timeout(time_limit, use_signals=False) def run_single_example(agent, env, example, max_steps, instruction, args, example_result_dir, scores, run): agent.reset() obs = env.reset(task_config=example)