Module Contents
API
-
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',
)[source]
Initialization
-
add_messages_key(
- example: dict[str, Any],
) → dict[str, list[dict[str, Any]]][source]
