Attempt to fix unittest issue

triton-inference-server · Jun 6, 2024 · c1acbe1 · c1acbe1
1 parent 3a07fc0
commit c1acbe1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)