Skip to content

Commit 1e80237

Browse files
committed
Fix naming for finalizer registry
Signed-off-by: Samuel Monson <[email protected]>
1 parent cb29da6 commit 1e80237

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/guidellm/benchmark/schemas/generative/entrypoints.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,7 @@ def get_default(cls: type[BenchmarkGenerativeTextArgs], field: str) -> Any:
215215
)
216216
)
217217
data_finalizer: DatasetFinalizer | str | dict[str, Any] = Field(
218-
default="generative_text_finalizer",
218+
default="generative",
219219
description="Finalizer for preparing data samples into requests",
220220
)
221221
data_collator: Callable | Literal["generative"] | None = Field(

src/guidellm/data/finalizers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ class FinalizerRegistry(RegistryMixin[type[DatasetFinalizer]]):
1919
pass
2020

2121

22-
@FinalizerRegistry.register("generative_request_finalizer")
22+
@FinalizerRegistry.register("generative")
2323
class GenerativeRequestFinalizer(DatasetFinalizer[GenerationRequest]):
2424
"""
2525
Finalizer that converts dataset rows into GenerationRequest objects,

0 commit comments

Comments
 (0)