Skip to content

Commit

Permalink
fix docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
fschlatt committed Nov 21, 2024
1 parent 3ba91b1 commit 52cf726
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions lightning_ir/cross_encoder/tokenizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,10 @@ def __init__(self, *args, query_length: int = 32, doc_length: int = 512, **kwarg
""":class:`.LightningIRTokenizer` for cross-encoder models. Encodes queries and documents jointly and ensures
that the input sequences are of the correct length.
:param query_length: _description_, defaults to 32
:param query_length: Maximum number of tokens per query, defaults to 32
:type query_length: int, optional
:param doc_length: _description_, defaults to 512
:param doc_length: Maximum number of tokens per document, defaults to 512
:type doc_length: int, optional
:type doc_length: int, optional
"""
super().__init__(*args, query_length=query_length, doc_length=doc_length, **kwargs)
Expand Down

0 comments on commit 52cf726

Please sign in to comment.