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]