Support Tuple observation spaces (openai#911)

Jorsorokin · May 31, 2019 · 21776e8 · 21776e8
1 parent 9b68103
commit 21776e8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/baselines/common/vec_env/util.py b/baselines/common/vec_env/util.py
@@ -38,6 +38,9 @@ def obs_space_info(obs_space):
     if isinstance(obs_space, gym.spaces.Dict):
         assert isinstance(obs_space.spaces, OrderedDict)
         subspaces = obs_space.spaces
+    elif isinstance(obs_space, gym.spaces.Tuple):
+        assert isinstance(obs_space.spaces, tuple)
+        subspaces = {i: obs_space.spaces[i] for i in range(len(obs_space.spaces))}
     else:
         subspaces = {None: obs_space}
     keys = []