Преглед на файлове

use new gym api for rendering in RGBImgObsWrapper

saleml преди 2 години
родител
ревизия
cf723609e4
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  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
         )