workaround openai/gym3#8

epignatelli · Jan 11, 2023 · 4698066 · 4698066
1 parent 3295547
commit 4698066
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/helx/environment/gym.py b/helx/environment/gym.py
@@ -52,14 +52,20 @@ def state(self) -> Array:
         return self._current_observation
 
     def reset(self, seed: int | None = None) -> Timestep:
-        obs, _ = self._env.reset(seed=seed)
+        try:
+            obs, _ = self._env.reset(seed=seed)
+        except TypeError:
+            # TODO(epignatelli): remove try/except when gym3 is updated.
+            # see: https://github.com/openai/gym3/issues/8
+            obs, _ = self._env.reset()
         self._current_observation = jnp.asarray(obs)
         return Timestep(obs, None, StepType.TRANSITION)
 
     def step(self, action: Action) -> Timestep:
         action_ = np.asarray(action)
         next_step = self._env.step(action_)
         self._current_observation = jnp.asarray(next_step[0])
+        gym.core.ObsType
         return Timestep.from_gym(next_step)
 
     def seed(self, seed: int) -> None: