Refactoring to late interaction class

qdrant · Nov 27, 2024 · e43f680 · e43f680
1 parent d581de9
commit e43f680
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/...action/late_interaction_embedding_base.py → ...n/late_interaction_text_embedding_base.py b/...action/late_interaction_embedding_base.py → ...n/late_interaction_text_embedding_base.py
@@ -42,9 +42,7 @@ def passage_embed(self, texts: Iterable[str], **kwargs) -> Iterable[np.ndarray]:
         # This is model-specific, so that different models can have specialized implementations
         yield from self.embed(texts, **kwargs)
 
-    def query_embed(
-        self, query: Union[str, Iterable[str]], **kwargs
-    ) -> Iterable[np.ndarray]:
+    def query_embed(self, query: Union[str, Iterable[str]], **kwargs) -> Iterable[np.ndarray]:
         """
         Embeds queries