From 66fd2cd5c33d1b9a26a8f3e8adef2e6ec1214868 Mon Sep 17 00:00:00 2001 From: bhack Date: Thu, 14 Mar 2024 21:59:54 +0100 Subject: [PATCH] Restore package_exists --- src/transformers/utils/import_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/utils/import_utils.py b/src/transformers/utils/import_utils.py index 027afc938a4..7fbfd950339 100644 --- a/src/transformers/utils/import_utils.py +++ b/src/transformers/utils/import_utils.py @@ -46,6 +46,7 @@ def _is_package_available(pkg_name: str, return_version: bool = False) -> Union[ try: # Primary method to get the package version package_version = importlib.metadata.version(pkg_name) + package_exists = True except importlib.metadata.PackageNotFoundError: # Fallback method: Only for "torch" and versions containing "dev" if pkg_name == "torch":