Fixing bugs for numpy 2.0 migration.

PiperOrigin-RevId: 665567692 Change-Id: I11d57a8ccb97171bac017f8f57a94c4710895278
tensorflow · Aug 20, 2024 · ef9d350 · ef9d350
1 parent 8662dab
commit ef9d350
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tf_agents/environments/atari_preprocessing_test.py b/tf_agents/environments/atari_preprocessing_test.py
@@ -55,7 +55,7 @@ def __init__(self, screen_size=10, max_steps=10):
     self.game_over = False
 
   def reset(self):
-    self.ale.screen_value = 10
+    self.ale.screen_value = 20
     self.num_steps = 0
     return self.get_observation()
 
@@ -118,7 +118,7 @@ def testMaxFramePooling(self):
 
     # The first observation is 2, the second 0; max is 2.
     observation, _, _, _ = env.step(0)
-    self.assertTrue((observation == 8).all())
+    self.assertTrue((observation == 18).all())
 
 
 if __name__ == '__main__':