renecotyfanboy / jaxspec

jaxspec is an X-ray spectra Bayesian analysis package, relying on JAX to enable just in time compilation
https://jaxspec.readthedocs.io/en/latest/
MIT License
29 stars 2 forks source link

Fix deprecation warnings from jax 0.28 #154

Closed renecotyfanboy closed 3 months ago

renecotyfanboy commented 5 months ago
/home/dups/miniconda3/envs/jaxspec/lib/python3.11/site-packages/sparse/_coo/core.py:216: DeprecationWarning: coords should be an ndarray. This will raise a ValueError in the future.
 /home/dups/miniconda3/envs/jaxspec/lib/python3.11/site-packages/jax/_src/linear_util.py:192: DeprecationWarning: Passing arguments 'a', 'a_min', or 'a_max' to jax.numpy.clip is deprecated. Please use 'x', 'min', and 'max' respectively instead.
  /home/dups/jaxspec/src/jaxspec/model/abc.py:132: DeprecationWarning: jax.tree_map is deprecated: use jax.tree.map (jax v0.4.25 or newer) or jax.tree_util.tree_map (any JAX version).