Merge pull request stanfordnlp#1145 from adam-simple/main

KNN few shot fixes

Author: arnavsinghvi11

dspy/teleprompt/knn_fewshot.py

@@ -9,20 +9,20 @@
 
 
 class KNNFewShot(Teleprompter):
-    def __init__(self, k: int, trainset: List[dsp.Example], vectorizer: Optional[dsp.BaseSentenceVectorizer] = None):
+    def __init__(self, k: int, trainset: List[dsp.Example], vectorizer: Optional[dsp.BaseSentenceVectorizer] = None, **few_shot_bootstrap_args):
         self.KNN = KNN(k, trainset, vectorizer=vectorizer)
+        self.few_shot_bootstrap_args = few_shot_bootstrap_args
 
     def compile(self, student, *, teacher=None, trainset=None, valset=None):
         student_copy = student.reset_copy()
 
-        def forward_pass(*args, **kwargs):
+        def forward_pass(_, **kwargs):
             knn_trainset = self.KNN(**kwargs)
-            few_shot_bootstrap = BootstrapFewShot()
+            few_shot_bootstrap = BootstrapFewShot(**self.few_shot_bootstrap_args)
             compiled_program = few_shot_bootstrap.compile(
                 student,
                 teacher=teacher,
                 trainset=knn_trainset,
-                valset=valset,
             )
             return compiled_program(**kwargs)