Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Bug: get_array_module fails when JAX is not installed #606

Closed
mrava87 opened this issue Aug 17, 2024 · 0 comments · Fixed by #607
Closed

Bug: get_array_module fails when JAX is not installed #606

mrava87 opened this issue Aug 17, 2024 · 0 comments · Fixed by #607
Labels
bug Something isn't working

Comments

@mrava87
Copy link
Collaborator

mrava87 commented Aug 17, 2024

This line:

if deps.cupy_enabled or deps.jax_enabled:
    if isinstance(x, jnp.ndarray):
        return jnp

raises an error if cupy is installed by not jax.

@mrava87 mrava87 added the bug Something isn't working label Aug 17, 2024
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
bug Something isn't working
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant