Bug: get_array_module fails when JAX is not installed

PyLops / pylops

PyLops – A Linear-Operator Library for Python

https://pylops.readthedocs.io

GNU Lesser General Public License v3.0

429 stars 102 forks source link

Closed mrava87 closed 3 months ago

mrava87 commented 3 months ago

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.