File tree Expand file tree Collapse file tree 1 file changed +8
-6
lines changed
onnxruntime/python/tools/transformers Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Original file line number Diff line number Diff line change 66# It is used to dump machine information for Notebooks
77
88import argparse
9+ import importlib .metadata
910import json
1011import logging
1112import platform
@@ -122,10 +123,7 @@ def get_gpu_info_by_nvml(self) -> dict:
122123 return result
123124
124125 def get_related_packages (self ) -> list [str ]:
125- import pkg_resources # noqa: PLC0415
126-
127- installed_packages = pkg_resources .working_set
128- related_packages = [
126+ related_packages = {
129127 "onnxruntime-gpu" ,
130128 "onnxruntime" ,
131129 "onnx" ,
@@ -137,8 +135,12 @@ def get_related_packages(self) -> list[str]:
137135 "flatbuffers" ,
138136 "numpy" ,
139137 "onnxconverter-common" ,
140- ]
141- related_packages_list = {i .key : i .version for i in installed_packages if i .key in related_packages }
138+ }
139+ related_packages_list = {}
140+ for dist in importlib .metadata .distributions ():
141+ if dist .metadata ["Name" ].lower () in related_packages :
142+ related_packages_list [dist .metadata ["Name" ].lower ()] = dist .version
143+
142144 return related_packages_list
143145
144146 def get_onnxruntime_info (self ) -> dict :
You can’t perform that action at this time.
0 commit comments