We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bbd93cd commit c4bfcdaCopy full SHA for c4bfcda
tests/test_dependencies.py
@@ -28,6 +28,7 @@
28
from packaging.requirements import Requirement
29
30
import lighteval.utils.imports as imports
31
+from lighteval.utils.imports import Extra
32
33
34
def pretend_missing(*names):
@@ -42,7 +43,11 @@ def wrapper(*args, **kwargs):
42
43
from unittest.mock import patch
44
45
def fake(requirement):
- name = requirement.name if isinstance(requirement, Requirement) else requirement
46
+ name = requirement
47
+ if isinstance(requirement, Requirement):
48
+ name = requirement.name
49
+ elif isinstance(requirement, Extra):
50
+ name = requirement.value
51
return False if name in names else (importlib.util.find_spec(name) is not None)
52
53
with patch.object(imports, "is_package_available", side_effect=fake):
0 commit comments