fix type checking

triton-inference-server · Jan 17, 2024 · 44dc20f · 44dc20f
1 parent bd8a9fc
commit 44dc20f
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_search_run_config_generator.py b/model_analyzer/config/generate/brute_plus_binary_search_run_config_generator.py
@@ -156,13 +156,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_inference_load(
         self, run_config: RunConfig, model_parameters: Dict, inference_load: 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
@@ -162,3 +162,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 {}