@@ -212,19 +212,16 @@ def cli():
212212@click .option (
213213 "--disable-progress" ,
214214 is_flag = True ,
215- default = not GenerativeTextScenario .model_fields ["show_progress" ].default ,
216215 help = "Set this flag to disable progress updates to the console" ,
217216)
218217@click .option (
219218 "--display-scheduler-stats" ,
220219 is_flag = True ,
221- default = GenerativeTextScenario .model_fields ["show_progress_scheduler_stats" ].default ,
222220 help = "Set this flag to display stats for the processes running the benchmarks" ,
223221)
224222@click .option (
225223 "--disable-console-outputs" ,
226224 is_flag = True ,
227- default = not GenerativeTextScenario .model_fields ["output_console" ].default ,
228225 help = "Set this flag to disable console output" ,
229226)
230227@click .option (
@@ -241,7 +238,6 @@ def cli():
241238@click .option (
242239 "--output-extras" ,
243240 callback = parse_json ,
244- default = GenerativeTextScenario .model_fields ["output_extras" ].default ,
245241 help = "A JSON string of extra data to save with the output benchmarks" ,
246242)
247243@click .option (
@@ -314,18 +310,18 @@ def benchmark(
314310 max_requests = max_requests ,
315311 warmup_percent = warmup_percent ,
316312 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 ,
322313 output_sampling = output_sampling ,
323314 random_seed = random_seed ,
324315 ))
325316
326317 asyncio .run (
327318 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 ,
329325 )
330326 )
331327
0 commit comments