`evorl.evaluators.ec_evaluator`¶

Module Contents¶

Classes¶

EpisodeObsCollector

Streamlined episode collector for observation only.

Functions¶

`env_step`
`rollout`

API¶

class evorl.evaluators.ec_evaluator.EpisodeObsCollector[source]¶

Bases: evorl.evaluators.episode_collector.EpisodeCollector

Streamlined episode collector for observation only.

rollout_fn: evorl.evaluators.episode_collector.RolloutFn¶: ‘pytree_field(…)’

evorl.evaluators.ec_evaluator.env_step(env_fn: evorl.envs.EnvStepFn, action_fn: evorl.agent.AgentActionFn, env_state: evorl.envs.EnvState, agent_state: evorl.agent.AgentState, key: chex.PRNGKey) → tuple[evorl.sample_batch.SampleBatch, evorl.envs.EnvState][source]¶

evorl.evaluators.ec_evaluator.rollout(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, env_extra_fields: collections.abc.Sequence[str] = ()) → tuple[evorl.sample_batch.SampleBatch, evorl.envs.EnvState][source]¶