Skip to content

Commit 8ce45b8

Browse files
committed
debug parallel task
1 parent 912932d commit 8ce45b8

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

experiments/run_osworld.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,16 +13,17 @@ def main():
1313
n_jobs = 1
1414
os.environ["AGENTLAB_DEBUG"] = "1"
1515
study = make_study(
16-
benchmark=OsworldBenchmark(test_set_name='test_small.json'), # type: ignore
16+
benchmark=OsworldBenchmark(test_set_name="test_small.json"), # type: ignore
1717
agent_args=[OSWORLD_CLAUDE],
18-
comment="osworld debug 1",
18+
comment="osworld debug 2",
1919
logging_level=logging.INFO,
2020
logging_level_stdout=logging.INFO,
2121
)
2222

2323
if os.environ.get("AGENTLAB_DEBUG"):
24-
study.exp_args_list = study.exp_args_list[-9:-8]
25-
study.run(n_jobs=1, n_relaunch=1, parallel_backend="sequential")
24+
study.exp_args_list = study.exp_args_list[-9:-7]
25+
print(f"Debug on {len(study.exp_args_list)} experiments")
26+
study.run(n_jobs=2, n_relaunch=1, parallel_backend="ray")
2627
else:
2728
study.run(n_jobs=n_jobs, n_relaunch=1, parallel_backend="ray")
2829

0 commit comments

Comments
 (0)