Skip to content

Commit 427f151

Browse files
sjmonsonAlonKellner-RedHat
authored andcommitted
Add list conversion for more fields
Signed-off-by: Samuel Monson <[email protected]> Signed-off-by: Alon Kellner <[email protected]>
1 parent 6f847cf commit 427f151

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/guidellm/benchmark/schemas.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1955,13 +1955,13 @@ def get_default(cls: type[BenchmarkGenerativeTextArgs], field: str) -> Any:
19551955
description="Whether to stop the benchmark if the model is over-saturated",
19561956
)
19571957

1958-
@field_validator("data", mode="wrap")
1958+
@field_validator("data", "data_args", "rate", mode="wrap")
19591959
@classmethod
19601960
def single_to_list(
19611961
cls, value: Any, handler: ValidatorFunctionWrapHandler
19621962
) -> list[Any]:
19631963
"""
1964-
Ensures 'data' field is always a list.
1964+
Ensures field is always a list.
19651965
19661966
:param value: Input value for the 'data' field
19671967
:return: List of data sources

0 commit comments

Comments
 (0)