Skip to content

Commit 4348f7e

Browse files
committed
fix type checking
1 parent 780b88f commit 4348f7e

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

model_analyzer/config/generate/brute_plus_binary_parameter_search_run_config_generator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,13 +150,13 @@ def _get_model_parameters(self, model_name: str) -> Dict:
150150
for model in self._models:
151151
if model_name == model.model_name():
152152
return model.parameters()
153-
154153
return {}
155154

156155
def _get_model_perf_analyzer_flags(self, model_name: str) -> Dict:
157156
for model in self._models:
158157
if model_name == model.model_name():
159158
return model.perf_analyzer_flags()
159+
return {}
160160

161161
def _set_parameter(
162162
self, run_config: RunConfig, model_parameters: Dict, parameter: int

model_analyzer/config/generate/quick_plus_concurrency_sweep_run_config_generator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -157,3 +157,4 @@ def _get_model_perf_analyzer_flags(self, model_name: str) -> Dict:
157157
for model in self._models:
158158
if model_name == model.model_name():
159159
return model.perf_analyzer_flags()
160+
return {}

0 commit comments

Comments
 (0)