SubprocVecEnv uses CloudpickleWrapper to send specs (#1028)

openai · Nov 8, 2019 · 391811d · 391811d
1 parent 665b888
commit 391811d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/baselines/common/vec_env/subproc_vec_env.py b/baselines/common/vec_env/subproc_vec_env.py
@@ -26,7 +26,7 @@ def step_env(env, action):
                 remote.close()
                 break
             elif cmd == 'get_spaces_spec':
-                remote.send((envs[0].observation_space, envs[0].action_space, envs[0].spec))
+                remote.send(CloudpickleWrapper((envs[0].observation_space, envs[0].action_space, envs[0].spec)))
             else:
                 raise NotImplementedError
     except KeyboardInterrupt:
@@ -68,7 +68,7 @@ def __init__(self, env_fns, spaces=None, context='spawn', in_series=1):
             remote.close()
 
         self.remotes[0].send(('get_spaces_spec', None))
-        observation_space, action_space, self.spec = self.remotes[0].recv()
+        observation_space, action_space, self.spec = self.remotes[0].recv().x
         self.viewer = None
         VecEnv.__init__(self, nenvs, observation_space, action_space)