evorl.envs.wrappers.obs_wrapper

Module Contents

Classes

ObsFlattenWrapper

Flatten the multi-dimention observation array into a 1D vector.

API

class evorl.envs.wrappers.obs_wrapper.ObsFlattenWrapper(env: evorl.envs.env.Env)

Bases: evorl.envs.wrappers.wrapper.Wrapper

Flatten the multi-dimention observation array into a 1D vector.

property obs_space: evorl.envs.space.Space

reset(key: chex.PRNGKey) → evorl.envs.env.EnvState

step(state: evorl.envs.env.EnvState, action: evorl.types.Action) → evorl.envs.env.EnvState