From 930f948164b645f835119d20dd31d87957544c59 Mon Sep 17 00:00:00 2001 From: "sweep-ai[bot]" <128439645+sweep-ai[bot]@users.noreply.github.com> Date: Wed, 20 Dec 2023 08:08:10 +0000 Subject: [PATCH] feat: Updated dspy/teleprompt/random_search.py --- dspy/teleprompt/random_search.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/dspy/teleprompt/random_search.py b/dspy/teleprompt/random_search.py index f027f83fe..5268c98cb 100644 --- a/dspy/teleprompt/random_search.py +++ b/dspy/teleprompt/random_search.py @@ -48,7 +48,7 @@ def __init__(self, metric, teacher_settings={}, max_bootstrapped_demos=4, max_la # print("Going to sample", self.max_num_traces, "traces in total.") print("Will attempt to train", self.num_candidate_sets, "candidate sets.") - def compile(self, student, *, teacher=None, trainset, valset=None, restrict=None): + async def compile(self, student, *, teacher=None, trainset, valset=None, restrict=None): self.trainset = trainset self.valset = valset or trainset # TODO: FIXME: Note this choice. @@ -70,14 +70,14 @@ def compile(self, student, *, teacher=None, trainset, valset=None, restrict=None elif seed == -2: # labels only teleprompter = LabeledFewShot(k=self.max_labeled_demos) - program2 = teleprompter.compile(student, trainset=trainset2) + program2 = await teleprompter.compile(student, trainset=trainset2) elif seed == -1: # unshuffled few-shot program = BootstrapFewShot(metric=self.metric, max_bootstrapped_demos=self.max_num_samples, max_labeled_demos=self.max_labeled_demos, teacher_settings=self.teacher_settings, max_rounds=self.max_rounds) - program2 = program.compile(student, teacher=teacher, trainset=trainset2) + program2 = await program.compile(student, teacher=teacher, trainset=trainset2) else: assert seed >= 0, seed @@ -95,7 +95,7 @@ def compile(self, student, *, teacher=None, trainset, valset=None, restrict=None evaluate = Evaluate(devset=self.valset, metric=self.metric, num_threads=self.num_threads, display_table=False, display_progress=True) - score, subscores = evaluate(program2, return_all_scores=True) + score, subscores = await evaluate(program2, return_all_scores=True) all_subscores.append(subscores)