added a n_repeat functionality to make_env_args_list_from_fixed_seeds() (#332)

Author: optimass

browsergym/experiments/src/browsergym/experiments/benchmark/utils.py

@@ -73,28 +73,29 @@ def make_env_args_list_from_repeat_tasks(
 
 
 def make_env_args_list_from_fixed_seeds(
-    task_list: list[str], max_steps: int, fixed_seeds: list[int]
+    task_list: list[str], max_steps: int, fixed_seeds: list[int], n_repeats: int = 1
 ):
     """
     Generates a list of `len(task_list)` time `n_repeats` environments arguments, using randomly generated seeds.
     """
     env_args_list = []
     for task in task_list:
         for seed in fixed_seeds:
-            env_args_list.append(
-                EnvArgs(
-                    task_name=task,
-                    task_seed=int(seed),
-                    max_steps=max_steps,
-                    headless=True,
-                    record_video=False,
-                    wait_for_user_message=False,
-                    viewport=None,
-                    slow_mo=None,
-                    storage_state=None,
-                    task_kwargs=None,
+            for _ in range(n_repeats):
+                env_args_list.append(
+                    EnvArgs(
+                        task_name=task,
+                        task_seed=int(seed),
+                        max_steps=max_steps,
+                        headless=True,
+                        record_video=False,
+                        wait_for_user_message=False,
+                        viewport=None,
+                        slow_mo=None,
+                        storage_state=None,
+                        task_kwargs=None,
+                    )
                 )
-            )
 
     return env_args_list