From 391811d98cfd1e4c7b5940d84a76ab7f423fa932 Mon Sep 17 00:00:00 2001 From: Tomasz Wrona Date: Sat, 9 Nov 2019 00:23:49 +0100 Subject: [PATCH] SubprocVecEnv uses CloudpickleWrapper to send specs (#1028) --- baselines/common/vec_env/subproc_vec_env.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/baselines/common/vec_env/subproc_vec_env.py b/baselines/common/vec_env/subproc_vec_env.py index 4a690c4..8c74000 100644 --- a/baselines/common/vec_env/subproc_vec_env.py +++ b/baselines/common/vec_env/subproc_vec_env.py @@ -26,7 +26,7 @@ def worker(remote, parent_remote, env_fn_wrappers): 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 @@ class SubprocVecEnv(VecEnv): 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)