gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

SubprocVecEnv uses CloudpickleWrapper to send specs (#1028)

Browse Source
This commit is contained in:
Tomasz Wrona
2019-11-09 00:23:49 +01:00
committed by pzhokhov
parent 665b888eeb
commit 391811d98c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
baselines/common/vec_env/subproc_vec_env.py
View File

@@ -26,7 +26,7 @@ def worker(remote, parent_remote, env_fn_wrappers):
remote.close() remote.close()
break break
elif cmd == 'get_spaces_spec': 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: else:
raise NotImplementedError raise NotImplementedError
except KeyboardInterrupt: except KeyboardInterrupt:
@@ -68,7 +68,7 @@ class SubprocVecEnv(VecEnv):
remote.close() remote.close()
self.remotes[0].send(('get_spaces_spec', None)) 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 self.viewer = None
VecEnv.__init__(self, nenvs, observation_space, action_space) VecEnv.__init__(self, nenvs, observation_space, action_space)