tinker_cookbook.distillation.SDFTBatchProvider
Generated from tinker-cookbook at commit 8486b8d. Source links point at that snapshot.
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]]