Wrappers

Environment wrappers compatible with the RLEnv protocol.

Episode Wrapper

class nnx_ppo.wrappers.episode_wrapper.EpisodeWrapper(env, max_len)

Bases: object

__init__(env, max_len)

step(state, action)

reset(rng)

property observation_size

property action_size

Reward Scaling Wrapper

class nnx_ppo.wrappers.reward_scaling_wrapper.RewardScalingWrapper(env, reward_scale)

Bases: object

__init__(env, reward_scale)

reset(rng)

step(state, action)

property observation_size: Any

property action_size: int