gxm.wrappers#

gxm.wrappers.Wrapper(env[, unwrap])

Base class for environment wrappers in gxm.

gxm.wrappers.Discretize(env, actions[, unwrap])

Wrapper that discretizes a continuous action space.

gxm.wrappers.FlattenObservation(env[, unwrap])

Wrapper that adds a rollout method to the environment.

gxm.wrappers.RecordEpisodeStatistics(env[, ...])

A wrapper that records the episode length \(T\) , episodic return \(J(\tau) = \sum_{t=0}^{T} r_t\) , and discounted episodic return \(G(\tau) = \sum_{t=0}^{T} \gamma^t r_t\) at the end of each episode.

gxm.wrappers.Rollout(env)

Wrapper that adds a rollout method to the environment.

gxm.wrappers.StackObservations(env, n_stack)

Wrapper that stacks the observation along a new axis.

gxm.wrappers.IgnoreTruncation(env, actions)

A wrapper that treats truncation as termination and removes the corresponding obsercation from the timestep.