`nemo_rl.data.datasets.response_datasets.geometry3k`#

Module Contents#

Classes#

Geometry3KDataset

Simple wrapper around the Geometry3K dataset.

Functions#

format_geometry3k_dataset

Format the Geometry3K dataset into an OpenAI-API-like message log.

API#

nemo_rl.data.datasets.response_datasets.geometry3k.format_geometry3k_dataset( example: dict[str, Any], return_pil: bool = False, ) → dict[str, Any]#: Format the Geometry3K dataset into an OpenAI-API-like message log.

class nemo_rl.data.datasets.response_datasets.geometry3k.Geometry3KDataset(split: str = 'train', **kwargs)#

Bases: nemo_rl.data.datasets.raw_dataset.RawDataset

Simple wrapper around the Geometry3K dataset.

Parameters:: split – Split name for the dataset, default is “train”

Initialization