瀏覽代碼

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
         )