Merge branch 'main' into fead/add-preprocess-dataset

Author: TomerG711

src/guidellm/__main__.py

@@ -32,7 +32,7 @@ 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."

src/guidellm/benchmark/entrypoints.py

@@ -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],