tinker_cookbook.distillation.SDFTBatchProvider

class tinker_cookbook.distillation.SDFTBatchProvider(Protocol)

Protocol for SDFT datasets that return builders alongside golden answers.

get_batch(index)

Return (env_group_builders, questions, golden_answers) for a batch.

Each list has the same length (one per problem in the batch).

Parameters:

• index (int)

Returns: tuple[Sequence[EnvGroupBuilder], list[str], list[str]]