From 386aa535f4e78c86db8877842f408df12126a73e Mon Sep 17 00:00:00 2001 From: Emanuele Fenocchi Date: Tue, 18 Jun 2024 22:32:03 +0000 Subject: [PATCH] fix(dspy): fixed formatting --- dspy/retrieve/deeplake_rm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dspy/retrieve/deeplake_rm.py b/dspy/retrieve/deeplake_rm.py index ae110b6c8..fa7fa9fe1 100644 --- a/dspy/retrieve/deeplake_rm.py +++ b/dspy/retrieve/deeplake_rm.py @@ -78,7 +78,7 @@ def embedding_function(self, texts, model="text-embedding-ada-002"): return [data.embedding for data in openai.embeddings.create(input = texts, model=model).data] def forward( - self, query_or_queries: Union[str, List[str]], k: Optional[int],**kwargs + self, query_or_queries: Union[str, List[str]], k: Optional[int],**kwargs, ) -> dspy.Prediction: """Search with DeepLake for self.k top passages for query