Skip to content

Commit b0e10bb

Browse files
committed
Make single-category curriculums the same as main curriculums
1 parent 2a5e864 commit b0e10bb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/browsergym/workarena/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,9 +149,10 @@ def get_all_tasks_agents(filter="l2", meta_seed=42, n_seed_l1=10, is_agent_curri
149149
ALL_COMPOSITIONAL_TASKS_CATEGORIES = HUMAN_CURRICULUM_L3
150150

151151
for category, items in ALL_COMPOSITIONAL_TASKS_CATEGORIES.items():
152+
category_seeds = rng.randint(0, 1000, items["num_seeds"])
152153
if filter_category and category != filter_category:
153154
continue
154-
for curr_seed in rng.randint(0, 1000, items["num_seeds"]):
155+
for curr_seed in category_seeds:
155156
random_gen = np.random.RandomState(curr_seed)
156157
for task_set, count in zip(items["buckets"], items["weights"]):
157158
tasks = random_gen.choice(task_set, count, replace=False)

0 commit comments

Comments
 (0)