From fae1195236fefb5bc04f87278477bb1f36bde330 Mon Sep 17 00:00:00 2001 From: tgerdes Date: Mon, 29 Jan 2024 14:48:07 -0600 Subject: [PATCH] fix precommit --- .../quick_plus_concurrency_sweep_run_config_generator.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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