fix: Fix type hint

hh-space-invader · hh-space-invader · commit bf5579306c0f · 2024-10-14T17:25:08.000+03:00
diff --git a/fastembed/late_interaction/colbert.py b/fastembed/late_interaction/colbert.py
@@ -221,7 +221,7 @@ def embed(
             **kwargs,
         )
 
-    def query_embed(self, query: Union[str, List[str]], **kwargs) -> np.ndarray:
+    def query_embed(self, query: Union[str, List[str]], **kwargs) -> Iterable[np.ndarray]:
         if isinstance(query, str):
             query = [query]
 

Original file line number	Diff line number	Diff line change
`@@ -221,7 +221,7 @@ def embed(`
`221`	`221`	`**kwargs,`
`222`	`222`	`)`
`223`	`223`
`224`		`- def query_embed(self, query: Union[str, List[str]], **kwargs) -> np.ndarray:`
	`224`	`+ def query_embed(self, query: Union[str, List[str]], **kwargs) -> Iterable[np.ndarray]:`
`225`	`225`	`if isinstance(query, str):`
`226`	`226`	`query = [query]`
`227`	`227`