biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Docs] vllm/benchmarks/datasets.py fix docstring param format. (#24970)

Browse Source 
Signed-off-by: samzong <samzong.lu@gmail.com>
This commit is contained in:
samzong
2025-09-17 23:11:51 +08:00
committed by GitHub
parent 8f3616f422
commit 4a2d33e371
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
vllm/benchmarks/datasets.py
View File

@@ -104,9 +104,9 @@ class BenchmarkDataset(ABC):
Args: Args:
dataset_path (Optional[str]): Path to the dataset. If None, it dataset_path (Optional[str]): Path to the dataset. If None, it
indicates that a default or random dataset might be used. indicates that a default or random dataset might be used.
random_seed (int): Seed value for reproducible shuffling or random_seed (int): Seed value for reproducible shuffling or
sampling. Defaults to DEFAULT_SEED. sampling. Defaults to DEFAULT_SEED.
""" """
self.dataset_path = dataset_path self.dataset_path = dataset_path
# Set the random seed, ensuring that a None value is replaced with the # Set the random seed, ensuring that a None value is replaced with the
@@ -200,8 +200,7 @@ class BenchmarkDataset(ABC):
tokenizer (PreTrainedTokenizerBase): The tokenizer to be used tokenizer (PreTrainedTokenizerBase): The tokenizer to be used
for processing the dataset's text. for processing the dataset's text.
num_requests (int): The number of sample requests to generate. num_requests (int): The number of sample requests to generate.
request_id_prefix (str) The prefix of request_id. request_id_prefix (str): The prefix of request_id.
Returns: Returns:
list[SampleRequest]: A list of sample requests generated from the list[SampleRequest]: A list of sample requests generated from the
@@ -224,7 +223,8 @@ class BenchmarkDataset(ABC):
requests (List[SampleRequest]): The current list of sampled requests (List[SampleRequest]): The current list of sampled
requests. requests.
num_requests (int): The target number of requests. num_requests (int): The target number of requests.
request_id_prefix (str) The prefix of the request ids. request_id_prefix (str): The prefix applied to generated request
identifiers.
""" """
if no_oversample: if no_oversample: