From b0e10bbdc39bb53eaa92f4857ef4121e9ceca76b Mon Sep 17 00:00:00 2001 From: Maxime Gasse Date: Mon, 7 Oct 2024 17:06:20 -0400 Subject: [PATCH 1/2] Make single-category curriculums the same as main curriculums --- src/browsergym/workarena/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/browsergym/workarena/__init__.py b/src/browsergym/workarena/__init__.py index 75ac279..f659cdb 100644 --- a/src/browsergym/workarena/__init__.py +++ b/src/browsergym/workarena/__init__.py @@ -149,9 +149,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) From 09d33998f1656c0199dd957e7f92633b5d52af16 Mon Sep 17 00:00:00 2001 From: Maxime Gasse Date: Tue, 8 Oct 2024 09:36:03 -0400 Subject: [PATCH 2/2] cleanup --- src/browsergym/workarena/__init__.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/browsergym/workarena/__init__.py b/src/browsergym/workarena/__init__.py index f659cdb..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: