Skip to content

Commit e86abf0

Browse files
committed
Fix default sweep setting
1 parent 26f3593 commit e86abf0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/guidellm/benchmark/profile.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
import numpy as np
3232
from pydantic import Field, computed_field, field_serializer, field_validator
3333

34+
from guidellm import settings
3435
from guidellm.scheduler import (
3536
AsyncConstantStrategy,
3637
AsyncPoissonStrategy,
@@ -620,7 +621,7 @@ def resolve_args(
620621
:param kwargs: Additional arguments to pass through.
621622
:return: Dictionary of resolved arguments.
622623
"""
623-
sweep_size_from_rate = rate[0] if rate else None
624+
sweep_size_from_rate = rate[0] if rate else settings.default_sweep_number
624625
kwargs["sweep_size"] = kwargs.get("sweep_size", sweep_size_from_rate)
625626
kwargs["random_seed"] = random_seed
626627
if rate_type in ["constant", "poisson"]:

0 commit comments

Comments
 (0)