diff --git a/PyTorch/Classification/GPUNet/triton/deployment_toolkit/triton_performance_runner/model_analyzer/runner.py b/PyTorch/Classification/GPUNet/triton/deployment_toolkit/triton_performance_runner/model_analyzer/runner.py index 53b27f8ed..47fcf0db6 100755 --- a/PyTorch/Classification/GPUNet/triton/deployment_toolkit/triton_performance_runner/model_analyzer/runner.py +++ b/PyTorch/Classification/GPUNet/triton/deployment_toolkit/triton_performance_runner/model_analyzer/runner.py @@ -29,16 +29,10 @@ from .model_analyzer import ModelAnalyzer, ModelAnalyzerMode from .model_analyzer_config import ModelAnalyzerConfig -if LooseVersion(sys.version) >= LooseVersion("3.8.0"): - from importlib.metadata import version +from importlib.metadata import version - TRITON_CLIENT_VERSION = LooseVersion(version("tritonclient")) - TRITON_MODEL_ANALYZER_VERSION = LooseVersion(version("triton-model-analyzer")) -else: - import pkg_resources - - TRITON_CLIENT_VERSION = LooseVersion(pkg_resources.get_distribution("tritonclient").version) - TRITON_MODEL_ANALYZER_VERSION = LooseVersion(pkg_resources.get_distribution("triton-model-analyzer").version) +TRITON_CLIENT_VERSION = LooseVersion(version("tritonclient")) +TRITON_MODEL_ANALYZER_VERSION = LooseVersion(version("triton-model-analyzer")) LOGGER = logging.getLogger("triton_performance_runner.model_analyzer") diff --git a/PyTorch/Classification/GPUNet/triton/deployment_toolkit/triton_performance_runner/perf_analyzer/runner.py b/PyTorch/Classification/GPUNet/triton/deployment_toolkit/triton_performance_runner/perf_analyzer/runner.py index 5dba64d91..d146d6faf 100755 --- a/PyTorch/Classification/GPUNet/triton/deployment_toolkit/triton_performance_runner/perf_analyzer/runner.py +++ b/PyTorch/Classification/GPUNet/triton/deployment_toolkit/triton_performance_runner/perf_analyzer/runner.py @@ -29,14 +29,9 @@ from .perf_analyzer import PerfAnalyzer from .perf_config import PerfAnalyzerConfig -if LooseVersion(sys.version) >= LooseVersion("3.8.0"): - from importlib.metadata import version +from importlib.metadata import version - TRITON_CLIENT_VERSION = LooseVersion(version("tritonclient")) -else: - import pkg_resources - - TRITON_CLIENT_VERSION = LooseVersion(pkg_resources.get_distribution("tritonclient").version) +TRITON_CLIENT_VERSION = LooseVersion(version("tritonclient")) LOGGER = logging.getLogger("triton_performance_runner.perf_analyzer") diff --git a/PyTorch/Forecasting/TFT/triton/run_performance_on_triton.py b/PyTorch/Forecasting/TFT/triton/run_performance_on_triton.py index cdaff5ee4..8c4213c9e 100644 --- a/PyTorch/Forecasting/TFT/triton/run_performance_on_triton.py +++ b/PyTorch/Forecasting/TFT/triton/run_performance_on_triton.py @@ -40,16 +40,10 @@ LOGGER = logging.getLogger("run_performance_on_triton") -if LooseVersion(sys.version) >= LooseVersion("3.8.0"): - from importlib.metadata import version +from importlib.metadata import version - TRITON_CLIENT_VERSION = LooseVersion(version("tritonclient")) - TRITON_MODEL_ANALYZER_VERSION = LooseVersion(version("triton-model-analyzer")) -else: - import pkg_resources - - TRITON_CLIENT_VERSION = LooseVersion(pkg_resources.get_distribution("tritonclient").version) - TRITON_MODEL_ANALYZER_VERSION = LooseVersion(pkg_resources.get_distribution("triton-model-analyzer").version) +TRITON_CLIENT_VERSION = LooseVersion(version("tritonclient")) +TRITON_MODEL_ANALYZER_VERSION = LooseVersion(version("triton-model-analyzer")) def _log_dict(title: str, dict_: Dict[str, Any]): diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/triton/run_performance_on_triton.py b/Tools/PyTorch/TimeSeriesPredictionPlatform/triton/run_performance_on_triton.py index 9c9526331..08869c5f3 100755 --- a/Tools/PyTorch/TimeSeriesPredictionPlatform/triton/run_performance_on_triton.py +++ b/Tools/PyTorch/TimeSeriesPredictionPlatform/triton/run_performance_on_triton.py @@ -40,16 +40,10 @@ LOGGER = logging.getLogger("run_performance_on_triton") -if LooseVersion(sys.version) >= LooseVersion("3.8.0"): - from importlib.metadata import version +from importlib.metadata import version - TRITON_CLIENT_VERSION = LooseVersion(version("tritonclient")) - TRITON_MODEL_ANALYZER_VERSION = LooseVersion(version("triton-model-analyzer")) -else: - import pkg_resources - - TRITON_CLIENT_VERSION = LooseVersion(pkg_resources.get_distribution("tritonclient").version) - TRITON_MODEL_ANALYZER_VERSION = LooseVersion(pkg_resources.get_distribution("triton-model-analyzer").version) +TRITON_CLIENT_VERSION = LooseVersion(version("tritonclient")) +TRITON_MODEL_ANALYZER_VERSION = LooseVersion(version("triton-model-analyzer")) def _log_dict(title: str, dict_: Dict[str, Any]):