Merge pull request #609 from PyLops/dev

Release v2.3.1

CHANGELOG.md

@@ -1,6 +1,9 @@
 Changelog
 =========
 
+# 2.3.1
+* Fixed bug in :py:mod:`pylops.utils.backend` (see [Issue #606](https:/PyLops/pylops/issues/606))
+
 # 2.3.0
 
 * Added `pylops.JaxOperator`, `pylops.signalprocessing.DWTND`, and `pylops.signalprocessing.DTCWT` operators.
@@ -20,9 +23,9 @@ Changelog
 * Removed `cusignal` as optional dependency and `cupy`'s equivalent methods (since the library
 is now unmantained and merged into `cupy`).
 * Fixed ImportError of optional dependencies when installed but not correctly functioning (see [Issue #548](https:/PyLops/pylops/issues/548))
-* Fixed bug in :py:func:`pylops.utils.deps.to_cupy_conditional` (see [Issue #579](https:/PyLops/pylops/issues/579))
-* Fixed bug in the definition of `nttot` in :py:class:`pylops.waveeqprocessing.BlendingContinuous`
-* Fixed bug in :py:func:`pylops.utils.signalprocessing.dip_estimate` (see [Issue #572](https:/PyLops/pylops/issues/572))
+* Fixed bug in `pylops.utils.deps.to_cupy_conditional` (see [Issue #579](https:/PyLops/pylops/issues/579))
+* Fixed bug in the definition of `nttot` in `pylops.waveeqprocessing.BlendingContinuous`
+* Fixed bug in `pylops.utils.signalprocessing.dip_estimate` (see [Issue #572](https:/PyLops/pylops/issues/572))
 
 
 # 2.2.0

docs/source/changelog.rst

@@ -3,6 +3,15 @@
 Changelog
 =========
 
+
+Version 2.3.1
+-------------
+
+*Released on: 17/08/2024*
+
+* Fixed bug in :py:mod:`pylops.utils.backend` (see https:/PyLops/pylops/issues/606)
+
+
 Version 2.3.0
 -------------
 

pylops/utils/backend.py

@@ -128,7 +128,7 @@ def get_array_module(x: npt.ArrayLike) -> ModuleType:
 
  """
  if deps.cupy_enabled or deps.jax_enabled:
- if isinstance(x, jnp.ndarray):
+ if deps.jax_enabled and isinstance(x, jnp.ndarray):
  return jnp
  elif deps.cupy_enabled:
  return cp.get_array_module(x)
@@ -153,7 +153,7 @@ def get_convolve(x: npt.ArrayLike) -> Callable:
 
  """
  if deps.cupy_enabled or deps.jax_enabled:
- if isinstance(x, jnp.ndarray):
+ if deps.jax_enabled and isinstance(x, jnp.ndarray):
  return j_convolve
  elif deps.cupy_enabled and cp.get_array_module(x) == cp:
  return cp_convolve
@@ -178,7 +178,7 @@ def get_fftconvolve(x: npt.ArrayLike) -> Callable:
 
  """
  if deps.cupy_enabled or deps.jax_enabled:
- if isinstance(x, jnp.ndarray):
+ if deps.jax_enabled and isinstance(x, jnp.ndarray):
  return j_fftconvolve
  elif deps.cupy_enabled and cp.get_array_module(x) == cp:
  return cp_fftconvolve
@@ -203,7 +203,7 @@ def get_oaconvolve(x: npt.ArrayLike) -> Callable:
 
  """
  if deps.cupy_enabled or deps.jax_enabled:
- if isinstance(x, jnp.ndarray):
+ if deps.jax_enabled and isinstance(x, jnp.ndarray):
  raise NotImplementedError(
  "oaconvolve not implemented in "
  "jax. Consider using a different"
@@ -232,7 +232,7 @@ def get_correlate(x: npt.ArrayLike) -> Callable:
 
  """
  if deps.cupy_enabled or deps.jax_enabled:
- if isinstance(x, jnp.ndarray):
+ if deps.jax_enabled and isinstance(x, jnp.ndarray):
  return jax.scipy.signal.correlate
  elif deps.cupy_enabled and cp.get_array_module(x) == cp:
  return cp_correlate
@@ -303,7 +303,7 @@ def get_block_diag(x: npt.ArrayLike) -> Callable:
 
  """
  if deps.cupy_enabled or deps.jax_enabled:
- if isinstance(x, jnp.ndarray):
+ if deps.jax_enabled and isinstance(x, jnp.ndarray):
  return jnp_block_diag
  elif deps.cupy_enabled and cp.get_array_module(x) == cp:
  return cp_block_diag
@@ -328,7 +328,7 @@ def get_toeplitz(x: npt.ArrayLike) -> Callable:
 
  """
  if deps.cupy_enabled or deps.jax_enabled:
- if isinstance(x, jnp.ndarray):
+ if deps.jax_enabled and isinstance(x, jnp.ndarray):
  return jnp_toeplitz
  elif deps.cupy_enabled and cp.get_array_module(x) == cp:
  return cp_toeplitz