chore: Changed query argument to Iterable to match base class

qdrant · Nov 13, 2024 · d0db6d8 · d0db6d8
1 parent 8c69952
commit d0db6d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastembed/late_interaction/colbert.py b/fastembed/late_interaction/colbert.py
@@ -232,7 +232,7 @@ def embed(
             **kwargs,
         )
 
-    def query_embed(self, query: Union[str, List[str]], **kwargs) -> Iterable[np.ndarray]:
+    def query_embed(self, query: Union[str, Iterable[str]], **kwargs) -> Iterable[np.ndarray]:
         if isinstance(query, str):
             query = [query]