nemo_rl.data.datasets.preference_datasets#

Submodules#

Package Contents#

Functions#

load_preference_dataset

Loads preference dataset.

Data#

DATASET_REGISTRY

__all__

API#

nemo_rl.data.datasets.preference_datasets.DATASET_REGISTRY#

None

nemo_rl.data.datasets.preference_datasets.load_preference_dataset(
data_config: nemo_rl.data.PreferenceDatasetConfig,
)#

Loads preference dataset.

Resolution order for data_config["dataset_name"]:

  1. If the name matches a key in DATASET_REGISTRY, use the built-in class.

  2. Otherwise, if the name contains a ., treat it as a fully qualified dotted import path (e.g. my_pkg.my_module.MyDataset) and import the class dynamically. This lets users register custom datasets without editing nemo_rl.

  3. Otherwise, raise ValueError with a helpful message.

nemo_rl.data.datasets.preference_datasets.__all__#

[‘BinaryPreferenceDataset’, ‘HelpSteer3Dataset’, ‘PreferenceDataset’, ‘Tulu3PreferenceDataset’, ‘loa…