tinker_cookbook.rl.RLDatasetBuilder

class tinker_cookbook.rl.RLDatasetBuilder(**)

Abstract builder that constructs training and optional test RL datasets.

__call__()

Build the training dataset and an optional test dataset.

Returns: tuple[RLDataset, RLDataset | None]: A two-element tuple where the first element is the training dataset and the second is an optional test dataset (None if no test set is needed).