mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-25 15:59:06 +00:00
ed8b13e11a
* Create transform_observation.py * Create test_transform_observation.py * Update __init__.py
27 lines
741 B
Python
27 lines
741 B
Python
from gym import ObservationWrapper
|
|
|
|
|
|
class TransformObservation(ObservationWrapper):
|
|
r"""Transform the observation via an arbitrary function.
|
|
|
|
Example::
|
|
|
|
>>> import gym
|
|
>>> env = gym.make('CartPole-v1')
|
|
>>> env = TransformObservation(env, lambda obs: obs + 0.1*np.random.randn(*obs.shape))
|
|
>>> env.reset()
|
|
array([-0.08319338, 0.04635121, -0.07394746, 0.20877492])
|
|
|
|
Args:
|
|
env (Env): environment
|
|
f (callable): a function that transforms the observation
|
|
|
|
"""
|
|
def __init__(self, env, f):
|
|
super(TransformObservation, self).__init__(env)
|
|
assert callable(f)
|
|
self.f = f
|
|
|
|
def observation(self, observation):
|
|
return self.f(observation)
|