`evorl.algorithms.ec.obs_utils`¶

Module Contents¶

Classes¶

ObsPreprocessor

Functions¶

`init_obs_preprocessor`
`init_obs_preprocessor_with_random_timesteps`
`rollout_obs`

API¶

class evorl.algorithms.ec.obs_utils.ObsPreprocessor[source]¶

Bases: evorl.types.PyTreeNode

random_timesteps: int¶: 0

static: bool¶: False

evorl.algorithms.ec.obs_utils.init_obs_preprocessor(agent_state, config, key, dp_axis_name=None)[source]¶

evorl.algorithms.ec.obs_utils.init_obs_preprocessor_with_random_timesteps(obs_preprocessor_state: Any, timesteps: int, env: evorl.envs.Env, key: chex.PRNGKey, dp_axis_name: str | None = None) → Any[source]¶

evorl.algorithms.ec.obs_utils.rollout_obs(env_fn: evorl.envs.EnvStepFn, action_fn: evorl.agent.AgentActionFn, env_state: evorl.envs.EnvState, agent_state: evorl.agent.AgentState, key: chex.PRNGKey, rollout_length: int) → tuple[chex.ArrayTree, evorl.envs.EnvState][source]¶