Explorar o código

use new gym api for rendering in RGBImgObsWrapper

saleml %!s(int64=2) %!d(string=hai) anos
pai
achega
cf723609e4
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      gym_minigrid/wrappers.py

+ 1 - 1
gym_minigrid/wrappers.py

@@ -174,9 +174,9 @@ class RGBImgObsWrapper(gym.core.ObservationWrapper):
 
     def observation(self, obs):
         env = self.unwrapped
+        assert env.render_mode == 'rgb_array', env.render_mode
 
         rgb_img = env.render(
-            mode='rgb_array',
             highlight=False,
             tile_size=self.tile_size
         )