Skip to content

Commit b5bac80

Browse files
committed
Refactor generate_benchmark_report to set default values for parameters
1 parent 535b5e9 commit b5bac80

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

src/guidellm/main.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -186,17 +186,17 @@ def generate_benchmark_report_cli(
186186

187187
def generate_benchmark_report(
188188
target: str,
189-
backend: BackendEnginePublic,
190-
model: Optional[str],
191189
data: Optional[str],
192190
data_type: Literal["emulated", "file", "transformers"],
193-
tokenizer: Optional[str],
194-
rate_type: ProfileGenerationMode,
195-
rate: Optional[float],
196-
max_seconds: Optional[int],
197-
max_requests: Union[Literal["dataset"], int, None],
198-
output_path: str,
199-
cont_refresh_table: bool,
191+
backend: BackendEnginePublic="openai_server",
192+
model: Optional[str]=None,
193+
tokenizer: Optional[str]=None,
194+
rate_type: ProfileGenerationMode="sweep",
195+
rate: Optional[float]=None,
196+
max_seconds: Optional[int]=120,
197+
max_requests: Union[Literal["dataset"], int, None]=None,
198+
output_path: str=None,
199+
cont_refresh_table: bool=False,
200200
) -> GuidanceReport:
201201
"""
202202
Generate a benchmark report for a specified backend and dataset.

0 commit comments

Comments
 (0)