format

Author: Hangzhi

dspy/teleprompt/bootstrap_finetune.py

@@ -83,7 +83,7 @@ def compile(
             data_pred_ind = None if self.multitask else pred_ind
             lm = pred.lm or settings.lm
             training_key = (lm, data_pred_ind)
-            
+
             if training_key not in key_to_data:
                 train_data, data_format = self._prepare_finetune_data(
                     trace_data=trace_data, lm=lm, pred_ind=data_pred_ind

tests/teleprompt/test_bootstrap_finetune.py

@@ -38,7 +38,7 @@ def forward(self, **kwargs):
 def test_compile_with_predict_instances_no_explicit_lm():
     """Test BootstrapFinetune compile with predictors that don't have explicit LMs."""
     from unittest.mock import patch
-    
+
     # Create student and teacher modules without explicit LMs in predictors
     student = SimpleModule("input -> output")
     teacher = SimpleModule("input -> output")
@@ -50,20 +50,20 @@ def test_compile_with_predict_instances_no_explicit_lm():
     # Verify that the predictor doesn't have an explicit LM
     assert student.predictor.lm is None
     bootstrap = BootstrapFinetune(metric=simple_metric)
-    
+
     # Mock all the components that would fail without proper setup
-    with patch('dspy.teleprompt.bootstrap_finetune.all_predictors_have_lms'), \
-            patch('dspy.teleprompt.bootstrap_finetune.prepare_teacher', return_value=teacher), \
-            patch('dspy.teleprompt.bootstrap_finetune.bootstrap_trace_data', return_value=[]), \
-            patch.object(bootstrap, '_prepare_finetune_data', return_value=([], 'openai')), \
-            patch.object(bootstrap, 'finetune_lms') as mock_finetune_lms:
-        
+    with patch("dspy.teleprompt.bootstrap_finetune.all_predictors_have_lms"), \
+            patch("dspy.teleprompt.bootstrap_finetune.prepare_teacher", return_value=teacher), \
+            patch("dspy.teleprompt.bootstrap_finetune.bootstrap_trace_data", return_value=[]), \
+            patch.object(bootstrap, "_prepare_finetune_data", return_value=([], "openai")), \
+            patch.object(bootstrap, "finetune_lms") as mock_finetune_lms:
+
         mock_finetune_lms.return_value = {(lm, None): lm}
-        
+
         # This should not raise AttributeError due to the fix
         compiled_student = bootstrap.compile(student, teacher=teacher, trainset=trainset)
-        
+
         assert compiled_student is not None, "Failed to compile student"
         mock_finetune_lms.assert_called_once()
-        
+