# {py:mod}`evorl.utils.ma_utils`

```{py:module} evorl.utils.ma_utils
```

```{autodoc2-docstring} evorl.utils.ma_utils
:parser: autodoc2_docstrings_parser
:allowtitles:
```

## Module Contents

### Functions

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

* - {py:obj}`batchify <evorl.utils.ma_utils.batchify>`
  - ```{autodoc2-docstring} evorl.utils.ma_utils.batchify
    :parser: autodoc2_docstrings_parser
    :summary:
    ```
* - {py:obj}`multi_agent_episode_done <evorl.utils.ma_utils.multi_agent_episode_done>`
  - ```{autodoc2-docstring} evorl.utils.ma_utils.multi_agent_episode_done
    :parser: autodoc2_docstrings_parser
    :summary:
    ```
* - {py:obj}`unbatchify <evorl.utils.ma_utils.unbatchify>`
  - ```{autodoc2-docstring} evorl.utils.ma_utils.unbatchify
    :parser: autodoc2_docstrings_parser
    :summary:
    ```
````

### API

````{py:function} batchify(x: dict[evorl.types.AgentID, jax.Array], agent_list, padding=False) -> jax.Array
:canonical: evorl.utils.ma_utils.batchify

```{autodoc2-docstring} evorl.utils.ma_utils.batchify
:parser: autodoc2_docstrings_parser
```
````

````{py:function} multi_agent_episode_done(done: evorl.types.Done) -> chex.Array
:canonical: evorl.utils.ma_utils.multi_agent_episode_done

```{autodoc2-docstring} evorl.utils.ma_utils.multi_agent_episode_done
:parser: autodoc2_docstrings_parser
```
````

````{py:function} unbatchify(x: jax.Array, agent_list) -> dict[evorl.types.AgentID, jax.Array]
:canonical: evorl.utils.ma_utils.unbatchify

```{autodoc2-docstring} evorl.utils.ma_utils.unbatchify
:parser: autodoc2_docstrings_parser
```
````