Skip to content

tinker_cookbook.rl.RLDataset

class tinker_cookbook.rl.RLDataset(ABC)

A dataset that produces batches of :class:EnvGroupBuilder instances.

get_batch(index)

Return a batch of EnvGroupBuilder instances at the given index.

Parameters:

  • index (int) – The batch index (0 <= index < len(self)).

Returns: Sequence[EnvGroupBuilder]: The environment group builders for this batch.