fixing test failures

This commit is contained in:
Peter Zhokhov
2018-10-31 14:10:52 -07:00
parent 63151af41a
commit 6fd2270c47

View File

@@ -20,8 +20,6 @@ class DummyVecEnv(VecEnv):
env = self.envs[0]
VecEnv.__init__(self, len(env_fns), env.observation_space, env.action_space)
obs_space = env.observation_space
if isinstance(obs_space, spaces.MultiDiscrete):
obs_space.shape = obs_space.shape[0]
self.keys, shapes, dtypes = obs_space_info(obs_space)