Browse Source

add condition to RGBImgObsWrapper's signature

saleml 2 years ago
parent
commit
162bab2fb5
1 changed files with 1 additions and 0 deletions
  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):