frozenleaves/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-10-20 17:13:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve package version check (#41661)

Browse Source 
fix
This commit is contained in:
Cyril Vallez
2025-10-16 17:31:58 +02:00
committed by GitHub
parent f7c33abab3
commit a408384a88
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/transformers/utils/import_utils.py
View File

@ -52,7 +52,10 @@ def _is_package_available(pkg_name: str, return_version: bool = False) -> Union[
try:
# importlib.metadata works with the distribution package, which may be different from the import
# name (e.g. `PIL` is the import name, but `pillow` is the distribution name)
distribution_name = PACKAGE_DISTRIBUTION_MAPPING[pkg_name][0]
distributions = PACKAGE_DISTRIBUTION_MAPPING[pkg_name]
# In most cases, the packages are well-behaved and both have the same name. If it's not the case, we
# pick the first item of the list as best guess (it's almost always a list of length 1 anyway)
distribution_name = pkg_name if pkg_name in distributions else distributions[0]
package_version = importlib.metadata.version(distribution_name)
except importlib.metadata.PackageNotFoundError:
# If we cannot find the metadata (because of editable install for example), try to import directly.