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 912e80c commit 4018fe4Copy full SHA for 4018fe4
tests/_helpers.py
@@ -6,8 +6,6 @@
6
7
8
def import_(library, wrapper=False):
9
- if library == 'cupy':
10
- return pytest.importorskip(library)
11
if 'jax' in library and sys.version_info < (3, 9):
12
pytest.skip('JAX array API support does not support Python 3.8')
13
@@ -16,5 +14,7 @@ def import_(library, wrapper=False):
16
14
library = 'jax.experimental.array_api'
17
15
else:
18
library = 'array_api_compat.' + library
+ elif library == 'cupy':
+ return pytest.importorskip(library)
19
20
return import_module(library)
0 commit comments