Skip to content

Commit c4bfcda

Browse files
committed
Adapt tests to Extras in requires
1 parent bbd93cd commit c4bfcda

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/test_dependencies.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
from packaging.requirements import Requirement
2929

3030
import lighteval.utils.imports as imports
31+
from lighteval.utils.imports import Extra
3132

3233

3334
def pretend_missing(*names):
@@ -42,7 +43,11 @@ def wrapper(*args, **kwargs):
4243
from unittest.mock import patch
4344

4445
def fake(requirement):
45-
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
4651
return False if name in names else (importlib.util.find_spec(name) is not None)
4752

4853
with patch.object(imports, "is_package_available", side_effect=fake):

0 commit comments

Comments
 (0)