From d0db6d8310abdeee5631c87bf179c17bffc44e07 Mon Sep 17 00:00:00 2001 From: hh-space-invader Date: Fri, 8 Nov 2024 15:27:28 +0200 Subject: [PATCH] chore: Changed query argument to Iterable to match base class --- fastembed/late_interaction/colbert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastembed/late_interaction/colbert.py b/fastembed/late_interaction/colbert.py index 41b58fb9..89b87805 100644 --- 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]