Merge branch 'main' into neurips-focus

ServiceNow · May 31, 2024 · 5bad698 · 5bad698
2 parents 40772e5 + fa768d8
commit 5bad698
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/experiments/src/browsergym/experiments/loop.py b/experiments/src/browsergym/experiments/loop.py
@@ -38,7 +38,7 @@ class EnvArgs:
     viewport: dict = None  # use default value from BrowserGym
     slow_mo: int = None  # use default value from BrowserGym
     storage_state: Optional[str | Path | dict] = None
-    task_kwargs: dict = field(default_factory=lambda: {})
+    task_kwargs: dict = None  # use default value from BrowserGym
 
     def make_env(self, action_mapping, exp_dir):
         extra_kwargs = {}
@@ -50,10 +50,11 @@ def make_env(self, action_mapping, exp_dir):
             extra_kwargs["slow_mo"] = self.slow_mo
         if self.storage_state:
             extra_kwargs["pw_context_kwargs"] = {"storage_state": self.storage_state}
+        if self.task_kwargs is not None:
+            extra_kwargs["task_kwargs"] = self.task_kwargs
 
         return gym.make(
             _get_env_name(self.task_name),
-            task_kwargs=self.task_kwargs,
             disable_env_checker=True,
             max_episode_steps=self.max_steps,
             headless=self.headless,