Skip to content

Commit b8580e6

Browse files
committed
Refactor backend preparation in Study class and improve logging for ignored dependencies
1 parent 418a05d commit b8580e6

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

src/agentlab/experiments/study.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -180,11 +180,7 @@ def _run(self, n_jobs=1, parallel_backend="joblib", strict_reproducibility=False
180180
raise ValueError("exp_args_list is None. Please set exp_args_list before running.")
181181

182182
logger.info("Preparing backends...")
183-
try:
184-
self.benchmark.prepare_backends()
185-
except ReadTimeout:
186-
logger.warning("Backend preparation timed out. Continuing anyway.")
187-
183+
self.benchmark.prepare_backends()
188184
logger.info("Backends ready.")
189185

190186
run_experiments(n_jobs, self.exp_args_list, self.dir, parallel_backend=parallel_backend)
@@ -382,8 +378,9 @@ def _agents_on_benchmark(
382378
# populate the depends_on field based on the task dependencies in the benchmark
383379
exp_args_list = add_dependencies(exp_args_list, benchmark.dependency_graph_over_tasks())
384380
else:
385-
logger.warning(f"Ignoring dependencies for benchmar {benchmark.name}. This could lead to different results.")
386-
381+
logger.warning(
382+
f"Ignoring dependencies for benchmar {benchmark.name}. This could lead to different results."
383+
)
387384

388385
return exp_args_list
389386

0 commit comments

Comments
 (0)