Changes by Brandon Johns. Bugfixes and removed use of latex.
requirements-cpu.txt
file because the JAX devs are in a hurry to deprecate code
- Removed use of latex from plots
- Added some more instructions to the readme
- Internal Bug
- In "example_DeLaN.py", "load_dataset()" had a mismatching number of output arguments
- Matplotlib deprecations 1
- Replace
fig.canvas.set_window_title
- with
fig.canvas.manager.set_window_title
- Replace
- Matplotlib deprecations 2
- Replace
cm.get_cmap
- with
matplotlib.colormaps.get_cmap
- and add
import matplotlib
to the top of each file that uses it
- Replace
- JAX deprecations 1
- See: jax-ml/jax#11280
- Replace
jax.partial
- with
functools.partial
- and add
import functools
to the top of each file that uses it
- JAX deprecations 2
- See: jax-ml/jax#11706
- Remove
mat_idx = jax.ops.index[..., mat_idx[0], mat_idx[1]]
- and
- Replace
triangular_mat = jax.ops.index_update(triangular_mat, mat_idx, vec_lower_triangular[:])
- with
triangular_mat = triangular_mat.at[mat_idx].set(vec_lower_triangular[:])
- JAX deprecations 3
- Replace
jax.tree_map
- with
jax.tree.map
- Replace