Skip to content

Commit 9ac84f3

Browse files
samzongchoprahetarth
authored andcommitted
[Docs] vllm/benchmarks/datasets.py fix docstring param format. (vllm-project#24970)
Signed-off-by: samzong <samzong.lu@gmail.com>
1 parent 158d8b5 commit 9ac84f3

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

vllm/benchmarks/datasets.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -104,9 +104,9 @@ def __init__(
104104
105105
Args:
106106
dataset_path (Optional[str]): Path to the dataset. If None, it
107-
indicates that a default or random dataset might be used.
107+
indicates that a default or random dataset might be used.
108108
random_seed (int): Seed value for reproducible shuffling or
109-
sampling. Defaults to DEFAULT_SEED.
109+
sampling. Defaults to DEFAULT_SEED.
110110
"""
111111
self.dataset_path = dataset_path
112112
# Set the random seed, ensuring that a None value is replaced with the
@@ -200,8 +200,7 @@ def sample(self, tokenizer: PreTrainedTokenizerBase,
200200
tokenizer (PreTrainedTokenizerBase): The tokenizer to be used
201201
for processing the dataset's text.
202202
num_requests (int): The number of sample requests to generate.
203-
request_id_prefix (str) The prefix of request_id.
204-
203+
request_id_prefix (str): The prefix of request_id.
205204
206205
Returns:
207206
list[SampleRequest]: A list of sample requests generated from the
@@ -224,7 +223,8 @@ def maybe_oversample_requests(
224223
requests (List[SampleRequest]): The current list of sampled
225224
requests.
226225
num_requests (int): The target number of requests.
227-
request_id_prefix (str) The prefix of the request ids.
226+
request_id_prefix (str): The prefix applied to generated request
227+
identifiers.
228228
229229
"""
230230
if no_oversample:

0 commit comments

Comments
 (0)