`nemo_rl.data`#

Subpackages#

nemo_rl.data.hf_datasets

Submodules#

Package Contents#

Classes#

`DataConfig`
`MathDataConfig`

API#

class nemo_rl.data.DataConfig[source]#

Bases: typing.TypedDict

max_input_seq_length: int#: None

prompt_file: str#: None

system_prompt_file: Optional[str]#: None

dataset_name: str#: None

val_dataset_name: Optional[str]#: None

add_bos: Optional[bool]#: None

add_eos: Optional[bool]#: None

input_key: Optional[str]#: None

output_key: Optional[str]#: None

class nemo_rl.data.MathDataConfig[source]#

Bases: nemo_rl.data.DataConfig

problem_key: str#: None

solution_key: str#: None