diff --git a/examples/atari/atari_wrapper.py b/examples/atari/atari_wrapper.py index 333f9787a..ad58ad231 100644 --- a/examples/atari/atari_wrapper.py +++ b/examples/atari/atari_wrapper.py @@ -24,7 +24,7 @@ def __init__(self, env, noop_max=30): def reset(self): self.env.reset() - noops = np.random.randint(1, self.noop_max + 1) + noops = self.unwrapped.np_random.randint(1, self.noop_max + 1) for _ in range(noops): obs, _, done, _ = self.env.step(self.noop_action) if done: