Skip to content

Commit eef95bc

Browse files
authored
Fix: Replace pkg_resources with importlib.metadata in machine_info.py (#27157)
Replaces the deprecated pkg_resources library with importlib.metadata to fix ModuleNotFoundError.
1 parent 0f153de commit eef95bc

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

onnxruntime/python/tools/transformers/machine_info.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
# It is used to dump machine information for Notebooks
77

88
import argparse
9+
import importlib.metadata
910
import json
1011
import logging
1112
import 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:

0 commit comments

Comments
 (0)