From f719298cf04a6362ee3582e9e7bc81fefa147fe9 Mon Sep 17 00:00:00 2001 From: Trevor Blackwell Date: Tue, 28 Mar 2017 10:50:48 -0700 Subject: [PATCH] Revert "Revert "Revert "Optimize Atari envs""" This reverts commit 8f62c0ab91fe28f95fd09fbe878252103e1d1771. --- gym/envs/atari/atari_env.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/gym/envs/atari/atari_env.py b/gym/envs/atari/atari_env.py index 040cac0e628..a304ccab20d 100644 --- a/gym/envs/atari/atari_env.py +++ b/gym/envs/atari/atari_env.py @@ -45,7 +45,7 @@ def __init__(self, game='pong', obs_type='ram', frameskip=(2, 5), repeat_action_ self._seed() (screen_width, screen_height) = self.ale.getScreenDims() - self._buffer = np.empty((screen_height, screen_width, 3), dtype=np.uint8) + self._buffer = np.empty((screen_height, screen_width, 4), dtype=np.uint8) self._action_set = self.ale.getMinimalActionSet() self.action_space = spaces.Discrete(len(self._action_set)) @@ -84,8 +84,8 @@ def _step(self, a): return ob, reward, self.ale.game_over(), {"ale.lives": self.ale.lives()} def _get_image(self): - self.ale.getScreenRGB2(self._buffer) # New in atari-py 2017-3-28 - return self._buffer + self.ale.getScreenRGB(self._buffer) # says rgb but actually bgr on little-endian systems like x86 + return self._buffer[:, :, [2, 1, 0]] def _get_ram(self): return to_ram(self.ale)