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

use env.render in dummy_vec_env.render when num_envs == 1 (#74)

Browse Source 
* use env.render in dummy_vec_env.render when num_envs == 1

* use shorter super() syntax per Alex's suggestion
This commit is contained in:
pzhokhov
2018-09-04 10:22:32 -07:00
committed by Peter Zhokhov
parent 51cefc933b
commit b3f966aa02
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
baselines/common/vec_env/dummy_vec_env.py
View File

@@ -66,3 +66,8 @@ class DummyVecEnv(VecEnv):
def get_images(self): def get_images(self):
return [env.render(mode='rgb_array') for env in self.envs] return [env.render(mode='rgb_array') for env in self.envs]
def render(self, mode='human'):
if self.num_envs == 1:
self.envs[0].render(mode=mode)
else:
super().render(mode=mode)