Skip to content

Commit 4cbf99a

Browse files
committed
Cleaner name getter
1 parent 98201ab commit 4cbf99a

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

tests/test_dependencies.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,9 @@
2727
from unittest.mock import patch
2828

2929
import pytest
30+
from packaging.requirements import Requirement
31+
32+
from lighteval.utils.imports import Extra
3033

3134

3235
def purge_lighteval():
@@ -48,7 +51,11 @@ def wrapper(*args, **kwargs):
4851
purge_lighteval()
4952

5053
def fake(requirement):
51-
name = getattr(requirement, "name", getattr(requirement, "value", requirement))
54+
name = requirement
55+
if isinstance(requirement, Requirement):
56+
name = requirement.name
57+
elif isinstance(requirement, Extra):
58+
name = requirement.value
5259

5360
# Probably a cleaner option here; returns False
5461
return False if name in names else (importlib.util.find_spec(name) is not None)

0 commit comments

Comments
 (0)