Skip to content

Commit

Permalink
Update fastembed/rerank/cross_encoder/text_cross_encoder_base.py
Browse files Browse the repository at this point in the history
Co-authored-by: George <[email protected]>
  • Loading branch information
I8dNLo and joein authored Dec 13, 2024
1 parent a05653f commit 1eb809b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fastembed/rerank/cross_encoder/text_cross_encoder_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ def rerank_pairs(
) -> Iterable[float]:
"""Rerank query-document pairs.
Args:
pairs Iterable[tuple[str, str]]: Query-document pairs to rerank
pairs (Iterable[tuple[str, str]]): Query-document pairs to rerank
batch_size (int): The batch size to use for reranking.
parallel: parallel:
If > 1, data-parallel encoding will be used, recommended for offline encoding of large datasets.
Expand Down

0 comments on commit 1eb809b

Please sign in to comment.