`evorl.envs.wrappers.wrapper`¶

Module Contents¶

Classes¶

Wraps an environment to allow modular transformations.

Functions¶

Return a specific wrapper of an env.

API¶

class evorl.envs.wrappers.wrapper.Wrapper(env: evorl.envs.env.Env)[source]¶

Bases: evorl.envs.env.Env

Wraps an environment to allow modular transformations.

property action_space: evorl.envs.env.Space¶

property obs_space: evorl.envs.env.Space¶

reset(key: chex.PRNGKey) → evorl.envs.env.EnvState[source]¶

step(state: evorl.envs.env.EnvState, action: evorl.types.Action) → evorl.envs.env.EnvState[source]¶

property unwrapped: evorl.envs.env.Env¶

evorl.envs.wrappers.wrapper.get_wrapper(env: evorl.envs.env.Env, wrapper_cls: type) → evorl.envs.wrappers.wrapper.Wrapper | None[source]¶: Return a specific wrapper of an env.