From c1acbe1198eaf1e1441e606a643984ffbe37c517 Mon Sep 17 00:00:00 2001 From: Brian Raf Date: Thu, 6 Jun 2024 17:00:44 +0000 Subject: [PATCH] Attempt to fix unittest issue --- model_analyzer/config/generate/search_parameters.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model_analyzer/config/generate/search_parameters.py b/model_analyzer/config/generate/search_parameters.py index 58f25e44c..f59104142 100755 --- a/model_analyzer/config/generate/search_parameters.py +++ b/model_analyzer/config/generate/search_parameters.py @@ -15,7 +15,7 @@ # limitations under the License. from math import log2 -from typing import Any, Dict, List, Optional, Tuple +from typing import Any, Dict, List, Optional, Tuple, Union from model_analyzer.config.input.config_command_profile import ConfigCommandProfile from model_analyzer.model_analyzer_exceptions import TritonModelAnalyzerException @@ -239,7 +239,7 @@ def _is_max_queue_delay_in_model_config_parameters(self) -> bool: def _populate_list_parameter( self, parameter_name: str, - parameter_list: List[int | str], + parameter_list: List[Union[int, str]], parameter_category: ParameterCategory, ) -> None: usage = self._determine_parameter_usage(parameter_name)