Skip to content

Commit 687f702

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

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)