Skip to content

Commit

Permalink
fix: Fix type hint
Browse files Browse the repository at this point in the history
  • Loading branch information
hh-space-invader committed Oct 14, 2024
1 parent ce0e544 commit bf55793
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fastembed/late_interaction/colbert.py
Original file line number Diff line number Diff line change
@@ -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]

0 comments on commit bf55793

Please sign in to comment.