@@ -212,19 +212,16 @@ def cli():
212
212
@click .option (
213
213
"--disable-progress" ,
214
214
is_flag = True ,
215
- default = not GenerativeTextScenario .model_fields ["show_progress" ].default ,
216
215
help = "Set this flag to disable progress updates to the console" ,
217
216
)
218
217
@click .option (
219
218
"--display-scheduler-stats" ,
220
219
is_flag = True ,
221
- default = GenerativeTextScenario .model_fields ["show_progress_scheduler_stats" ].default ,
222
220
help = "Set this flag to display stats for the processes running the benchmarks" ,
223
221
)
224
222
@click .option (
225
223
"--disable-console-outputs" ,
226
224
is_flag = True ,
227
- default = not GenerativeTextScenario .model_fields ["output_console" ].default ,
228
225
help = "Set this flag to disable console output" ,
229
226
)
230
227
@click .option (
@@ -241,7 +238,6 @@ def cli():
241
238
@click .option (
242
239
"--output-extras" ,
243
240
callback = parse_json ,
244
- default = GenerativeTextScenario .model_fields ["output_extras" ].default ,
245
241
help = "A JSON string of extra data to save with the output benchmarks" ,
246
242
)
247
243
@click .option (
@@ -314,18 +310,18 @@ def benchmark(
314
310
max_requests = max_requests ,
315
311
warmup_percent = warmup_percent ,
316
312
cooldown_percent = cooldown_percent ,
317
- show_progress = not disable_progress ,
318
- show_progress_scheduler_stats = display_scheduler_stats ,
319
- output_console = not disable_console_outputs ,
320
- output_path = output_path ,
321
- output_extras = output_extras ,
322
313
output_sampling = output_sampling ,
323
314
random_seed = random_seed ,
324
315
))
325
316
326
317
asyncio .run (
327
318
benchmark_with_scenario (
328
- scenario = GenerativeTextScenario (** _scenario )
319
+ scenario = GenerativeTextScenario (** _scenario ),
320
+ show_progress = not disable_progress ,
321
+ show_progress_scheduler_stats = display_scheduler_stats ,
322
+ output_console = not disable_console_outputs ,
323
+ output_path = output_path ,
324
+ output_extras = output_extras ,
329
325
)
330
326
)
331
327
0 commit comments