File tree Expand file tree Collapse file tree 1 file changed +3
-9
lines changed
PyTorch/Forecasting/TFT/triton Expand file tree Collapse file tree 1 file changed +3
-9
lines changed Original file line number Diff line number Diff line change 4040
4141LOGGER = 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
5549def _log_dict (title : str , dict_ : Dict [str , Any ]):
You can’t perform that action at this time.
0 commit comments