Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions src/guidellm/__main__.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,10 +31,10 @@ def parse_number_str(ctx, param, value): # noqa: ARG001
values = value.split(",") if "," in value else [value]

try:
return [int(val) if val.isdigit() else float(val) for val in values]
return [float(val) for val in values]
except ValueError as err:
raise click.BadParameter(
f"{param.name} must be a number or comma-separated list of numbers."
f"{param.name} must be a floating-point number or comma-separated list of floating-point numbers."
) from err


Expand Down
2 changes: 1 addition & 1 deletion src/guidellm/benchmark/entrypoints.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ async def benchmark_generative_text(
data_args: Optional[dict[str, Any]],
data_sampler: Optional[Literal["random"]],
rate_type: Union[StrategyType, ProfileType],
rate: Optional[Union[int, float, list[Union[int, float]]]],
rate: Optional[Union[float, list[float]]],
max_seconds: Optional[float],
max_requests: Optional[int],
warmup_percent: Optional[float],
Expand Down
Loading