baselines: fix dummy vec env render mode (#42)

2018-08-26 20:54:38 -07:00
parent 1ea5ec647c
commit 34af61a132
1 changed files with 1 additions and 1 deletions
--- a/baselines/common/vec_env/dummy_vec_env.py
+++ b/baselines/common/vec_env/dummy_vec_env.py
@@ -64,5 +64,5 @@ class DummyVecEnv(VecEnv):
        return dict_to_obs(copy_obs_dict(self.buf_obs))

    def get_images(self):
-        return [env.render(mode='rgb') for env in self.envs]
+        return [env.render(mode='rgb_array') for env in self.envs]