diff --git a/src/diffusers/utils/testing_utils.py b/src/diffusers/utils/testing_utils.py index 7a524e76f16e..b01ca98b60dd 100644 --- a/src/diffusers/utils/testing_utils.py +++ b/src/diffusers/utils/testing_utils.py @@ -555,10 +555,10 @@ def decorator(test_case): def require_hf_hub_version_greater(hf_hub_version): def decorator(test_case): correct_hf_hub_version = version.parse( - version.parse(importlib.metadata.version("huggingface_hub")).base_version + version.parse(importlib.metadata.version("huggingface-hub")).base_version ) > version.parse(hf_hub_version) return unittest.skipUnless( - correct_hf_hub_version, f"Test requires huggingface_hub with the version greater than {hf_hub_version}." + correct_hf_hub_version, f"Test requires huggingface-hub with the version greater than {hf_hub_version}." )(test_case) return decorator