bugfix: when using configure, update observation space

f1tenth · Apr 5, 2024 · 83355aa · 83355aa
1 parent 7ed8521
commit 83355aa
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/gym/f110_gym/envs/f110_env.py b/gym/f110_gym/envs/f110_env.py
@@ -268,6 +268,12 @@ def configure(self, config: dict) -> None:
                     self.action_type.space, self.num_agents
                 )
 
+            if hasattr(self, "observation_space"):
+                self.observation_type = observation_factory(
+                    env=self, **self.config["observation_config"]
+                )
+                self.observation_space = self.observation_type.space()
+
     def _check_done(self):
         """Check if the current rollout is done