Переглянути джерело

add condition to RGBImgObsWrapper's signature

saleml 3 роки тому
батько
коміт
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,
     Wrapper to use fully observable RGB image as observation,
     This can be used to have the agent to solve the gridworld in pixel space.
     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):
     def __init__(self, env, tile_size=8):