Update return type in TextEmbeddingBase

qdrant · Feb 7, 2024 · bb724f0 · bb724f0
1 parent a69f1f2
commit bb724f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastembed/text/text_embedding_base.py b/fastembed/text/text_embedding_base.py
@@ -47,7 +47,7 @@ def query_embed(self, query: Union[str, Iterable[str]], **kwargs) -> Iterable[np
             query (Union[str, Iterable[str]]): The query to embed, or an iterable e.g. list of queries.
 
         Returns:
-            np.ndarray: The embeddings.
+            Iterable[np.ndarray]: The embeddings.
         """
 
         # This is model-specific, so that different models can have specialized implementations