Skip to content

Commit 4018fe4

Browse files
committed
Fix import_('cupy', wrapper=True) tests helper
1 parent 912e80c commit 4018fe4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/_helpers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,6 @@
66

77

88
def import_(library, wrapper=False):
9-
if library == 'cupy':
10-
return pytest.importorskip(library)
119
if 'jax' in library and sys.version_info < (3, 9):
1210
pytest.skip('JAX array API support does not support Python 3.8')
1311

@@ -16,5 +14,7 @@ def import_(library, wrapper=False):
1614
library = 'jax.experimental.array_api'
1715
else:
1816
library = 'array_api_compat.' + library
17+
elif library == 'cupy':
18+
return pytest.importorskip(library)
1919

2020
return import_module(library)

0 commit comments

Comments
 (0)