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.

Parameters:

• index (int)