nemo_rl.data.datasets.eval_datasets.aime

AIME dataset.

Module Contents

Classes

AIMEDataset

Data

AIMEVariant

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])