Browse Source

use new gym api for rendering in RGBImgObsWrapper

saleml 2 years ago
parent
commit
cf723609e4
1 changed files with 1 additions and 1 deletions
  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):
     def observation(self, obs):
         env = self.unwrapped
         env = self.unwrapped
+        assert env.render_mode == 'rgb_array', env.render_mode
 
 
         rgb_img = env.render(
         rgb_img = env.render(
-            mode='rgb_array',
             highlight=False,
             highlight=False,
             tile_size=self.tile_size
             tile_size=self.tile_size
         )
         )