Skip to content

Commit 6684e3d

Browse files
authored
switching to 2 for loops in _agents_on_benchmark (#107)
1 parent f8d1e47 commit 6684e3d

File tree

1 file changed

+20
-8
lines changed

1 file changed

+20
-8
lines changed

src/agentlab/experiments/study.py

Lines changed: 20 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -358,14 +358,26 @@ def _agents_on_benchmark(
358358
if demo_mode:
359359
set_demo_mode(env_args_list)
360360

361-
exp_args_list = args.expand_cross_product(
362-
ExpArgs(
363-
agent_args=args.CrossProd(agents),
364-
env_args=args.CrossProd(env_args_list),
365-
logging_level=logging_level,
366-
logging_level_stdout=logging_level_stdout,
367-
)
368-
) # type: list[ExpArgs]
361+
# exp_args_list = args.expand_cross_product(
362+
# ExpArgs(
363+
# agent_args=args.CrossProd(agents),
364+
# env_args=args.CrossProd(env_args_list),
365+
# logging_level=logging_level,
366+
# logging_level_stdout=logging_level_stdout,
367+
# )
368+
# ) # type: list[ExpArgs]
369+
370+
exp_args_list = []
371+
372+
for agent in agents:
373+
for env_args in env_args_list:
374+
exp_args = ExpArgs(
375+
agent_args=agent,
376+
env_args=env_args,
377+
logging_level=logging_level,
378+
logging_level_stdout=logging_level_stdout,
379+
)
380+
exp_args_list.append(exp_args)
369381

370382
for i, exp_args in enumerate(exp_args_list):
371383
exp_args.order = i

0 commit comments

Comments
 (0)