`nemo_rl.data.datasets.eval_datasets.aime`#

AIME dataset.

Module Contents#

Classes#

Data#

API#

nemo_rl.data.datasets.eval_datasets.aime.AIMEVariant#: None

class nemo_rl.data.datasets.eval_datasets.aime.AIMEDataset( variant: nemo_rl.data.datasets.eval_datasets.aime.AIMEVariant = '2025', prompt_file: Optional[str] = None, system_prompt_file: Optional[str] = None, )#

Initialization

_rekey(data: dict[str, Any])#