瀏覽代碼

add condition to RGBImgObsWrapper's signature

saleml 2 年之前
父節點
當前提交
162bab2fb5
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      gym_minigrid/wrappers.py

+ 1 - 0
gym_minigrid/wrappers.py

@@ -156,6 +156,7 @@ class RGBImgObsWrapper(gym.core.ObservationWrapper):
     """
     Wrapper to use fully observable RGB image as observation,
     This can be used to have the agent to solve the gridworld in pixel space.
+    To use it, make the unwrapped environment with render_mode='rgb_array'.
     """
 
     def __init__(self, env, tile_size=8):