Skip to content

Commit 398dce6

Browse files
committed
Add list conversion for more fields
Signed-off-by: Samuel Monson <[email protected]>
1 parent 3148467 commit 398dce6

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/guidellm/benchmark/schemas.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1678,7 +1678,7 @@ def compile(
16781678
estimated_state: EstimatedBenchmarkState,
16791679
scheduler_state: SchedulerState,
16801680
profile: Profile,
1681-
requests: Iterable,
1681+
requests: Iterable, # noqa: ARG003
16821682
backend: BackendInterface,
16831683
environment: Environment,
16841684
strategy: SchedulingStrategy,
@@ -1845,7 +1845,7 @@ def get_default(cls: type[BenchmarkGenerativeTextArgs], field: str) -> Any:
18451845
profile: StrategyType | ProfileType | Profile = Field(
18461846
default="sweep", description="Benchmark profile or scheduling strategy type"
18471847
)
1848-
rate: float | list[float] | None = Field(
1848+
rate: list[float] | None = Field(
18491849
default=None, description="Request rate(s) for rate-based scheduling"
18501850
)
18511851
# Backend configuration
@@ -1938,13 +1938,13 @@ def get_default(cls: type[BenchmarkGenerativeTextArgs], field: str) -> Any:
19381938
default=None, description="Maximum global error rate (0-1) before stopping"
19391939
)
19401940

1941-
@field_validator("data", mode="wrap")
1941+
@field_validator("data", "data_args", "rate", mode="wrap")
19421942
@classmethod
19431943
def single_to_list(
19441944
cls, value: Any, handler: ValidatorFunctionWrapHandler
19451945
) -> list[Any]:
19461946
"""
1947-
Ensures 'data' field is always a list.
1947+
Ensures field is always a list.
19481948
19491949
:param value: Input value for the 'data' field
19501950
:return: List of data sources

0 commit comments

Comments
 (0)