Support Tuple observation spaces (#911)

2019-05-31 17:06:20 -04:00
parent 9b68103b73
commit 21776e8f57
1 changed files with 3 additions and 0 deletions
--- 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 = []