diff --git a/src/browsergym/workarena/__init__.py b/src/browsergym/workarena/__init__.py index 75ac279..df9c115 100644 --- a/src/browsergym/workarena/__init__.py +++ b/src/browsergym/workarena/__init__.py @@ -112,7 +112,6 @@ def get_task_category(task_name): def get_all_tasks_agents(filter="l2", meta_seed=42, n_seed_l1=10, is_agent_curriculum=True): - OFFSET = 42 all_task_tuples = [] filter = filter.split(".") if len(filter) > 2: @@ -149,9 +148,10 @@ def get_all_tasks_agents(filter="l2", meta_seed=42, n_seed_l1=10, is_agent_curri ALL_COMPOSITIONAL_TASKS_CATEGORIES = HUMAN_CURRICULUM_L3 for category, items in ALL_COMPOSITIONAL_TASKS_CATEGORIES.items(): + category_seeds = rng.randint(0, 1000, items["num_seeds"]) if filter_category and category != filter_category: continue - for curr_seed in rng.randint(0, 1000, items["num_seeds"]): + for curr_seed in category_seeds: random_gen = np.random.RandomState(curr_seed) for task_set, count in zip(items["buckets"], items["weights"]): tasks = random_gen.choice(task_set, count, replace=False)