Skip to content

Commit 17f469b

Browse files
sjmonsonAlonKellner-RedHat
authored andcommitted
Support dashed arguments for benchmark args
Signed-off-by: Samuel Monson <smonson@redhat.com> Signed-off-by: Alon Kellner <akellner@redhat.com>
1 parent df37695 commit 17f469b

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

src/guidellm/benchmark/schemas.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@
2424

2525
import yaml
2626
from pydantic import (
27+
AliasChoices,
28+
AliasGenerator,
2729
ConfigDict,
2830
Field,
2931
ValidationError,
@@ -1885,12 +1887,7 @@ def get_default(cls: type[BenchmarkGenerativeTextArgs], field: str) -> Any:
18851887
data_request_formatter: DatasetPreprocessor | dict[str, str] | str = Field(
18861888
default="chat_completions",
18871889
description="Request formatting preprocessor or template name",
1888-
validation_alias=AliasChoices(
1889-
"data_request_formatter",
1890-
"data-request-formatter",
1891-
"request_type",
1892-
"request-type",
1893-
),
1890+
validation_alias=AliasChoices("request_type", "request-type"),
18941891
)
18951892
data_collator: Callable | Literal["generative"] | None = Field(
18961893
default="generative", description="Data collator for batch processing"

0 commit comments

Comments
 (0)