# {py:mod}`evorl.evaluators.episode_collector`

```{py:module} evorl.evaluators.episode_collector
```

```{autodoc2-docstring} evorl.evaluators.episode_collector
:parser: autodoc2_docstrings_parser
:allowtitles:
```

## Module Contents

### Classes

````{list-table}
:class: autosummary longtable
:align: left

* - {py:obj}`EpisodeCollector <evorl.evaluators.episode_collector.EpisodeCollector>`
  - ```{autodoc2-docstring} evorl.evaluators.episode_collector.EpisodeCollector
    :parser: autodoc2_docstrings_parser
    :summary:
    ```
````

### API

`````{py:class} EpisodeCollector
:canonical: evorl.evaluators.episode_collector.EpisodeCollector

Bases: {py:obj}`evorl.types.PyTreeNode`

```{autodoc2-docstring} evorl.evaluators.episode_collector.EpisodeCollector
:parser: autodoc2_docstrings_parser
```

````{py:attribute} action_fn
:canonical: evorl.evaluators.episode_collector.EpisodeCollector.action_fn
:type: evorl.agent.AgentActionFn
:value: >
   None

```{autodoc2-docstring} evorl.evaluators.episode_collector.EpisodeCollector.action_fn
:parser: autodoc2_docstrings_parser
```

````

````{py:attribute} discount
:canonical: evorl.evaluators.episode_collector.EpisodeCollector.discount
:type: float
:value: >
   1.0

```{autodoc2-docstring} evorl.evaluators.episode_collector.EpisodeCollector.discount
:parser: autodoc2_docstrings_parser
```

````

````{py:attribute} env
:canonical: evorl.evaluators.episode_collector.EpisodeCollector.env
:type: evorl.envs.Env
:value: >
   None

```{autodoc2-docstring} evorl.evaluators.episode_collector.EpisodeCollector.env
:parser: autodoc2_docstrings_parser
```

````

````{py:attribute} env_extra_fields
:canonical: evorl.evaluators.episode_collector.EpisodeCollector.env_extra_fields
:type: collections.abc.Sequence[str]
:value: >
   ()

```{autodoc2-docstring} evorl.evaluators.episode_collector.EpisodeCollector.env_extra_fields
:parser: autodoc2_docstrings_parser
```

````

````{py:attribute} max_episode_steps
:canonical: evorl.evaluators.episode_collector.EpisodeCollector.max_episode_steps
:type: int
:value: >
   'pytree_field(...)'

```{autodoc2-docstring} evorl.evaluators.episode_collector.EpisodeCollector.max_episode_steps
:parser: autodoc2_docstrings_parser
```

````

````{py:method} rollout(agent_state, key: chex.PRNGKey, num_episodes: int) -> tuple[evorl.metrics.EvaluateMetric, evorl.sample_batch.SampleBatch]
:canonical: evorl.evaluators.episode_collector.EpisodeCollector.rollout

```{autodoc2-docstring} evorl.evaluators.episode_collector.EpisodeCollector.rollout
:parser: autodoc2_docstrings_parser
```

````

````{py:attribute} rollout_fn
:canonical: evorl.evaluators.episode_collector.EpisodeCollector.rollout_fn
:type: evorl.rollout.RolloutFn
:value: >
   'pytree_field(...)'

```{autodoc2-docstring} evorl.evaluators.episode_collector.EpisodeCollector.rollout_fn
:parser: autodoc2_docstrings_parser
```

````

`````