fix type checking

triton-inference-server · Jan 26, 2024 · 4348f7e · 4348f7e
1 parent 780b88f
commit 4348f7e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/model_analyzer/config/generate/brute_plus_binary_parameter_search_run_config_generator.py b/model_analyzer/config/generate/brute_plus_binary_parameter_search_run_config_generator.py
@@ -150,13 +150,13 @@ def _get_model_parameters(self, model_name: str) -> Dict:
         for model in self._models:
             if model_name == model.model_name():
                 return model.parameters()
-
         return {}
 
     def _get_model_perf_analyzer_flags(self, model_name: str) -> Dict:
         for model in self._models:
             if model_name == model.model_name():
                 return model.perf_analyzer_flags()
+        return {}
 
     def _set_parameter(
         self, run_config: RunConfig, model_parameters: Dict, parameter: int

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
@@ -157,3 +157,4 @@ def _get_model_perf_analyzer_flags(self, model_name: str) -> Dict:
         for model in self._models:
             if model_name == model.model_name():
                 return model.perf_analyzer_flags()
+        return {}