Skip to content

Commit c030718

Browse files
authored
Fix docstring to mention 'rerank' method (#648)
Updated docstring to reflect the correct method exposed by the model.
1 parent fcb951c commit c030718

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

libs/infinity_emb/infinity_emb/inference/batch_handler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,7 @@ async def rerank(
191191
if top_n is None, <= 0 or out of range, all scores are returned
192192
193193
Raises:
194-
ModelNotDeployedError: If loaded model does not expose `embed`
194+
ModelNotDeployedError: If loaded model does not expose `rerank`
195195
capabilities
196196
197197
Returns:

0 commit comments

Comments
 (0)