Use self.env instead of defining self._env

tensorflow · Aug 3, 2019 · c051e5f · c051e5f
1 parent 89b2dc1
commit c051e5f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tf_agents/environments/atari_preprocessing.py b/tf_agents/environments/atari_preprocessing.py
@@ -72,7 +72,6 @@ def __init__(self,
       ValueError: if frame_skip or screen_size are not strictly positive.
     """
     super(AtariPreprocessing, self).__init__(env)
-    self._env = env
 
     # Return the observation space adjusted to match the shape of the processed
     # observations.
@@ -111,7 +110,7 @@ def reset(self):
       observation: numpy array, the initial observation emitted by the
         environment.
     """
-    self._env.reset()
+    self.env.reset()
     self.lives = self.env.ale.lives()
     self.game_over = False
     self._fetch_grayscale_observation(self.screen_buffer[0])
@@ -144,7 +143,7 @@ def step(self, action):
     for time_step in range(self.frame_skip):
       # We bypass the Gym observation altogether and directly fetch the
       # grayscale image from the ALE. This is a little faster.
-      _, reward, game_over, info = self._env.step(action)
+      _, reward, game_over, info = self.env.step(action)
       accumulated_reward += reward
 
       if self.terminal_on_life_loss: