`nemo_rl.data.datasets.response_datasets.aime24`#

Module Contents#

Classes#

AIME2024Dataset

Simple wrapper around the AIME2024 dataset with train split.

API#

class nemo_rl.data.datasets.response_datasets.aime24.AIME2024Dataset(repeat: int = 16, **kwargs)#

Bases: nemo_rl.data.datasets.raw_dataset.RawDataset

Simple wrapper around the AIME2024 dataset with train split.

Parameters:: repeat – Number of times to repeat the dataset, default is 16

Initialization

format_data(data: dict[str, Any]) → dict[str, Any]#