Pārlūkot izejas kodu

add condition to RGBImgObsWrapper's signature

saleml 2 gadi atpakaļ
vecāks
revīzija
162bab2fb5
1 mainītis faili ar 1 papildinājumiem un 0 dzēšanām
  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):