diff --git a/model_analyzer/config/generate/quick_plus_concurrency_sweep_run_config_generator.py b/model_analyzer/config/generate/quick_plus_concurrency_sweep_run_config_generator.py index 35913ed11..1a0c50cd7 100755 --- a/model_analyzer/config/generate/quick_plus_concurrency_sweep_run_config_generator.py +++ b/model_analyzer/config/generate/quick_plus_concurrency_sweep_run_config_generator.py @@ -140,7 +140,9 @@ def _sweep_concurrency_over_top_results(self) -> Generator[RunConfig, None, None for result in top_results: run_config = deepcopy(result.run_config()) perf_analyzer_flags = self._get_model_perf_analyzer_flags(model_name) - parameter_search = ParameterSearch(self._config, perf_analyzer_flags=perf_analyzer_flags) + parameter_search = ParameterSearch( + self._config, perf_analyzer_flags=perf_analyzer_flags + ) for concurrency in parameter_search.search_parameters(): run_config = self._set_concurrency(run_config, concurrency) yield run_config