fix: Fix wrapper handling in gymnasium v1.0.0

strakam · Nov 23, 2024 · aaf4d43 · aaf4d43
1 parent 6504aa0
commit aaf4d43
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/generals/envs/gymnasium_wrappers.py b/generals/envs/gymnasium_wrappers.py
@@ -16,6 +16,7 @@ class ObservationAsImageWrapper(gym.ObservationWrapper):
     def __init__(self, env):
         super().__init__(env)
         n_obs_keys = len(self.observation_space["observation"].items())
+        self.game = env.game
         self.observation_space = gym.spaces.Dict(
             {
                 "observation": gym.spaces.Box(