Skip to content

Commit

Permalink
fix precommit
Browse files Browse the repository at this point in the history
  • Loading branch information
tgerdesnv committed Jan 29, 2024
1 parent 38abbe8 commit fae1195
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit fae1195

Please sign in to comment.