From 44dc20f045402752c9c103f0c3224fa7d8475914 Mon Sep 17 00:00:00 2001 From: tgerdes Date: Wed, 17 Jan 2024 15:21:09 -0600 Subject: [PATCH] fix type checking --- .../generate/brute_plus_binary_search_run_config_generator.py | 2 +- .../quick_plus_concurrency_sweep_run_config_generator.py | 1 + 2 files changed, 2 insertions(+), 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 index 0d55cb8ec..1b89f129b 100755 --- 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 index 141badaaf..9ab07ff77 100755 --- 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 {}