AtariEnvFactory: Fix super call

opcode81 · opcode81 · commit a86e246cfee1 · 2025-06-19T10:34:46.000+02:00
diff --git a/examples/atari/atari_wrapper.py b/examples/atari/atari_wrapper.py
@@ -414,7 +414,7 @@ def __init__(
         )
 
     def _create_env(self, mode: EnvMode) -> gym.Env:
-        env = super().create_env(mode)
+        env = super()._create_env(mode)
         is_train = mode == EnvMode.TRAIN
         return wrap_deepmind(
             env,

Original file line number	Diff line number	Diff line change
`@@ -414,7 +414,7 @@ def __init__(`
`414`	`414`	`)`
`415`	`415`
`416`	`416`	`def _create_env(self, mode: EnvMode) -> gym.Env:`
`417`		`- env = super().create_env(mode)`
	`417`	`+ env = super()._create_env(mode)`
`418`	`418`	`is_train = mode == EnvMode.TRAIN`
`419`	`419`	`return wrap_deepmind(`
`420`	`420`	`env,`