diff --git a/src/envs/wrappers.py b/src/envs/wrappers.py index b1054a1..eca6f88 100644 --- a/src/envs/wrappers.py +++ b/src/envs/wrappers.py @@ -64,7 +64,7 @@ class NoopResetEnv(gym.Wrapper): 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) + noops = self.unwrapped.np_random.integers(1, self.noop_max + 1) assert noops > 0 obs = None for _ in range(noops):