> For clean Markdown of any page, append .md to the page URL.
> For a complete documentation index, see https://fr.nvidia-localization.ferndocs.com/fr/llms.txt.
> For AI client integration (Claude Code, Cursor, etc.), connect to the MCP server at https://fr.nvidia-localization.ferndocs.com/fr/_mcp/server.

# nemo\_rl.data.hf\_datasets.prompt\_response\_dataset

## Contenu du module

### Classes

| Nom                                                                                        | Description |
| ------------------------------------------------------------------------------------------ | ----------- |
| [`PromptResponseDataset`](#nemorldatahfdatasetspromptresponsedatasetpromptresponsedataset) | Aucun       |

### API

```python
class nemo_rl.data.hf_datasets.prompt_response_dataset.PromptResponseDataset(train_ds_path: str, val_ds_path: str, input_key: str = 'input', output_key: str = 'output')
```

```python
add_messages_key(example: dict[str, typing.Any]) -> dict[str, list[dict[str, typing.Any]]]
```