Skip to content

Commit cb4bf2e

Browse files
Fix/non integer rates (#116)
This resolves a bug of only integer rates being supported. via the CLI `--rate` option. Adds support float rate values.
1 parent 8238b90 commit cb4bf2e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/guidellm/benchmark/profile.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -253,9 +253,9 @@ def from_standard_args( # type: ignore[override]
253253
if not isinstance(rate, Sequence):
254254
rate = [rate]
255255

256-
if not all(r.is_integer() and r > 0 for r in rate):
256+
if not all(isinstance(r, (float, int)) and r > 0 for r in rate):
257257
raise ValueError(
258-
f"All rate values must be positive integers, received {rate}"
258+
f"All rate values must be positive numbers, received {rate}"
259259
)
260260

261261
if rate_type == "async":

0 commit comments

Comments
 (0)