Enable flattening mixed discrete/continuous observation spaces

proroklab · Aug 1, 2023 · 24f42fb · 24f42fb
1 parent 041a9c6
commit 24f42fb
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/popgym/wrappers/flatten.py b/popgym/wrappers/flatten.py
@@ -34,7 +34,6 @@ def __init__(
                     self.env.action_space
                 )
         if flatten_observation:
-            self.continuous(self.env.observation_space)
             self.need_flatten_obs = isinstance(
                 self.env.observation_space, gym.spaces.Tuple
             ) or isinstance(self.env.observation_space, gym.spaces.Dict)