Merge pull request #174 from jfrancis71/master

Fix for issue #173
pfnet · Sep 21, 2022 · 2ad3d51 · 2ad3d51
2 parents d89ddf6 + f09e108
commit 2ad3d51
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pfrl/wrappers/atari_wrappers.py b/pfrl/wrappers/atari_wrappers.py
@@ -7,6 +7,7 @@
 import gym
 import numpy as np
 from gym import spaces
+from packaging import version
 
 import pfrl
 
@@ -37,9 +38,10 @@ def reset(self, **kwargs):
         if self.override_num_noops is not None:
             noops = self.override_num_noops
         else:
-            noops = self.unwrapped.np_random.randint(
-                1, self.noop_max + 1
-            )  # pylint: disable=E1101
+            if version.parse(gym.__version__) >= version.parse("0.24.0"):
+                noops = self.unwrapped.np_random.integers(1, self.noop_max + 1)
+            else:
+                noops = self.unwrapped.np_random.randint(1, self.noop_max + 1)
         assert noops > 0
         obs = None
         for _ in range(noops):