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 4cbf99a commit 02609dcCopy full SHA for 02609dc
tests/test_dependencies.py
@@ -27,9 +27,6 @@
27
from unittest.mock import patch
28
29
import pytest
30
-from packaging.requirements import Requirement
31
-
32
-from lighteval.utils.imports import Extra
33
34
35
def purge_lighteval():
@@ -52,10 +49,8 @@ def wrapper(*args, **kwargs):
52
49
53
50
def fake(requirement):
54
51
name = requirement
55
- if isinstance(requirement, Requirement):
56
- name = requirement.name
57
- elif isinstance(requirement, Extra):
58
- name = requirement.value
+ name = getattr(requirement, "name", name) # if it is a Requirement
+ name = getattr(requirement, "value", name) # if it is an Extra
59
60
# Probably a cleaner option here; returns False
61
return False if name in names else (importlib.util.find_spec(name) is not None)
0 commit comments