Browse Source

Update wrappers.py

Maxime Chevalier-Boisvert 3 years ago
parent
commit
9a94b5ebb9
1 changed files with 3 additions and 4 deletions
  1. 3 4
      gym_minigrid/wrappers.py

+ 3 - 4
gym_minigrid/wrappers.py

@@ -154,9 +154,8 @@ class OneHotPartialObsWrapper(gym.core.ObservationWrapper):
 
 
 class RGBImgObsWrapper(gym.core.ObservationWrapper):
 class RGBImgObsWrapper(gym.core.ObservationWrapper):
     """
     """
-    Wrapper to use fully observable RGB image as the only observation output,
-    no language/mission. This can be used to have the agent to solve the
-    gridworld in pixel space.
+    Wrapper to use fully observable RGB image as observation,
+    This can be used to have the agent to solve the gridworld in pixel space.
     """
     """
 
 
     def __init__(self, env, tile_size=8):
     def __init__(self, env, tile_size=8):
@@ -188,7 +187,7 @@ class RGBImgObsWrapper(gym.core.ObservationWrapper):
 
 
 class RGBImgPartialObsWrapper(gym.core.ObservationWrapper):
 class RGBImgPartialObsWrapper(gym.core.ObservationWrapper):
     """
     """
-    Wrapper to use partially observable RGB image as the only observation output
+    Wrapper to use partially 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.
     """
     """