Skip to content

Commit 44dc20f

Browse files
committed
fix type checking
1 parent bd8a9fc commit 44dc20f

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

model_analyzer/config/generate/brute_plus_binary_search_run_config_generator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,13 +156,13 @@ def _get_model_parameters(self, model_name: str) -> Dict:
156156
for model in self._models:
157157
if model_name == model.model_name():
158158
return model.parameters()
159-
160159
return {}
161160

162161
def _get_model_perf_analyzer_flags(self, model_name: str) -> Dict:
163162
for model in self._models:
164163
if model_name == model.model_name():
165164
return model.perf_analyzer_flags()
165+
return {}
166166

167167
def _set_inference_load(
168168
self, run_config: RunConfig, model_parameters: Dict, inference_load: 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
@@ -162,3 +162,4 @@ def _get_model_perf_analyzer_flags(self, model_name: str) -> Dict:
162162
for model in self._models:
163163
if model_name == model.model_name():
164164
return model.perf_analyzer_flags()
165+
return {}

0 commit comments

Comments
 (0)