Skip to content

Commit e556f4f

Browse files
committed
fix: remove pkg_resources fallback for setuptools 82+ compatibility
1 parent 729963d commit e556f4f

1 file changed

Lines changed: 3 additions & 9 deletions

File tree

  • PyTorch/Classification/GPUNet/triton/deployment_toolkit/triton_performance_runner/model_analyzer

PyTorch/Classification/GPUNet/triton/deployment_toolkit/triton_performance_runner/model_analyzer/runner.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -29,16 +29,10 @@
2929
from .model_analyzer import ModelAnalyzer, ModelAnalyzerMode
3030
from .model_analyzer_config import ModelAnalyzerConfig
3131

32-
if LooseVersion(sys.version) >= LooseVersion("3.8.0"):
33-
from importlib.metadata import version
32+
from importlib.metadata import version
3433

35-
TRITON_CLIENT_VERSION = LooseVersion(version("tritonclient"))
36-
TRITON_MODEL_ANALYZER_VERSION = LooseVersion(version("triton-model-analyzer"))
37-
else:
38-
import pkg_resources
39-
40-
TRITON_CLIENT_VERSION = LooseVersion(pkg_resources.get_distribution("tritonclient").version)
41-
TRITON_MODEL_ANALYZER_VERSION = LooseVersion(pkg_resources.get_distribution("triton-model-analyzer").version)
34+
TRITON_CLIENT_VERSION = LooseVersion(version("tritonclient"))
35+
TRITON_MODEL_ANALYZER_VERSION = LooseVersion(version("triton-model-analyzer"))
4236

4337
LOGGER = logging.getLogger("triton_performance_runner.model_analyzer")
4438

0 commit comments

Comments
 (0)