diff --git a/auto_round/utils.py b/auto_round/utils.py index c63367ea..60ca6d99 100644 --- a/auto_round/utils.py +++ b/auto_round/utils.py @@ -833,14 +833,14 @@ def get_library_version(library_name): if python_vesion < Version("3.8"): import warnings warnings.filterwarnings('ignore', category=DeprecationWarning) - import pkg_resources + import pkg_resources # pylint: disable=E0401 try: version = pkg_resources.get_distribution(library_name).version return version except pkg_resources.DistributionNotFound: return f"{library_name} is not installed" else: - import importlib_metadata + import importlib_metadata # pylint: disable=E0401 try: version = importlib_metadata.version(library_name) return version