|
@@ -126,7 +126,6 @@ class RGBImgObsWrapper(gym.core.ObservationWrapper):
|
|
self.__dict__.update(vars(env)) # Pass values to super wrapper
|
|
self.__dict__.update(vars(env)) # Pass values to super wrapper
|
|
super().__init__(env)
|
|
super().__init__(env)
|
|
|
|
|
|
-
|
|
|
|
self.observation_space = spaces.Box(
|
|
self.observation_space = spaces.Box(
|
|
low=0,
|
|
low=0,
|
|
high=255,
|
|
high=255,
|
|
@@ -138,7 +137,6 @@ class RGBImgObsWrapper(gym.core.ObservationWrapper):
|
|
env = self.unwrapped
|
|
env = self.unwrapped
|
|
return env.render(mode = 'rgb_array', highlight = False)
|
|
return env.render(mode = 'rgb_array', highlight = False)
|
|
|
|
|
|
-
|
|
|
|
class FullyObsWrapper(gym.core.ObservationWrapper):
|
|
class FullyObsWrapper(gym.core.ObservationWrapper):
|
|
"""
|
|
"""
|
|
Fully observable gridworld using a compact grid encoding
|
|
Fully observable gridworld using a compact grid encoding
|