nemo_rl.data.eval_datasets.aime2024

AIME 2024 dataset.

Module Contents

Classes

AIME2024Dataset

API

class nemo_rl.data.eval_datasets.aime2024.AIME2024Dataset(

prompt_file: Optional[str] = None,

system_prompt_file: Optional[str] = None,

)

Initialization

_rekey(data: dict[str, Any])