From 21776e8f5776efd1ac297946d25dc4c202f1dcdc Mon Sep 17 00:00:00 2001
From: Joshua Meier <jmeier@fb.com>
Date: Fri, 31 May 2019 17:06:20 -0400
Subject: [PATCH] Support Tuple observation spaces (#911)

---
 baselines/common/vec_env/util.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/baselines/common/vec_env/util.py b/baselines/common/vec_env/util.py
index d29d8a3..ddc71c2 100644
--- 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 = []