Skip to content

Commit 6bb876f

Browse files
committed
addressed PR comments
1 parent 794cf99 commit 6bb876f

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/diffusers/utils/import_utils.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from collections import OrderedDict
2323
from itertools import chain
2424
from types import ModuleType
25-
from typing import Any, Union
25+
from typing import Any, Tuple, Union
2626

2727
from huggingface_hub.utils import is_jinja_available # noqa: F401
2828
from packaging.version import Version, parse
@@ -54,7 +54,7 @@
5454
_is_google_colab = "google.colab" in sys.modules or any(k.startswith("COLAB_") for k in os.environ)
5555

5656

57-
def _is_package_available(pkg_name: str, get_dist_name: bool = False) -> tuple[bool, str]:
57+
def _is_package_available(pkg_name: str, get_dist_name: bool = False) -> Tuple[bool, str]:
5858
pkg_exists = importlib.util.find_spec(pkg_name) is not None
5959
pkg_version = "N/A"
6060

@@ -63,7 +63,9 @@ def _is_package_available(pkg_name: str, get_dist_name: bool = False) -> tuple[b
6363
package_map = importlib_metadata.packages_distributions()
6464
if get_dist_name and pkg_name in package_map and package_map[pkg_name]:
6565
if len(package_map[pkg_name]) > 1:
66-
logger.warning(f"Multiple distributions found for package {pkg_name}. Picked distribution: {package_map[pkg_name][0]}")
66+
logger.warning(
67+
f"Multiple distributions found for package {pkg_name}. Picked distribution: {package_map[pkg_name][0]}"
68+
)
6769
pkg_name = package_map[pkg_name][0]
6870
pkg_version = importlib_metadata.version(pkg_name)
6971
logger.debug(f"Successfully imported {pkg_name} version {pkg_version}")

0 commit comments

Comments
 (0)