Explorar el Código

add condition to RGBImgObsWrapper's signature

saleml hace 2 años
padre
commit
162bab2fb5
Se han modificado 1 ficheros con 1 adiciones y 0 borrados
  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):