Skip to content

Commit ff31762

Browse files
committed
fix: remove pkg_resources fallback for setuptools 82+ compatibility
1 parent 6ae74e7 commit ff31762

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

Tools/PyTorch/TimeSeriesPredictionPlatform/triton/run_performance_on_triton.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -40,16 +40,10 @@
4040

4141
LOGGER = logging.getLogger("run_performance_on_triton")
4242

43-
if LooseVersion(sys.version) >= LooseVersion("3.8.0"):
44-
from importlib.metadata import version
43+
from importlib.metadata import version
4544

46-
TRITON_CLIENT_VERSION = LooseVersion(version("tritonclient"))
47-
TRITON_MODEL_ANALYZER_VERSION = LooseVersion(version("triton-model-analyzer"))
48-
else:
49-
import pkg_resources
50-
51-
TRITON_CLIENT_VERSION = LooseVersion(pkg_resources.get_distribution("tritonclient").version)
52-
TRITON_MODEL_ANALYZER_VERSION = LooseVersion(pkg_resources.get_distribution("triton-model-analyzer").version)
45+
TRITON_CLIENT_VERSION = LooseVersion(version("tritonclient"))
46+
TRITON_MODEL_ANALYZER_VERSION = LooseVersion(version("triton-model-analyzer"))
5347

5448

5549
def _log_dict(title: str, dict_: Dict[str, Any]):

0 commit comments

Comments
 (0)