dockyard_rl.data.datasets.response_datasets.gemometry3k

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

dockyard_rl.data.datasets.response_datasets.gemometry3k.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 dockyard_rl.data.datasets.response_datasets.gemometry3k.Geometry3KDataset(split: str = 'train', **kwargs)

Bases: dockyard_rl.data.datasets.raw_dataset.RawDataset

Simple wrapper around the Geometry3K dataset.

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

Initialization