Skip to content

Commit 02609dc

Browse files
committed
Quick fix
1 parent 4cbf99a commit 02609dc

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

tests/test_dependencies.py

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

2929
import pytest
30-
from packaging.requirements import Requirement
31-
32-
from lighteval.utils.imports import Extra
3330

3431

3532
def purge_lighteval():
@@ -52,10 +49,8 @@ def wrapper(*args, **kwargs):
5249

5350
def fake(requirement):
5451
name = requirement
55-
if isinstance(requirement, Requirement):
56-
name = requirement.name
57-
elif isinstance(requirement, Extra):
58-
name = requirement.value
52+
name = getattr(requirement, "name", name) # if it is a Requirement
53+
name = getattr(requirement, "value", name) # if it is an Extra
5954

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

0 commit comments

Comments
 (0)