pyro-ppl / numpyro / 22040729202

93%

push

github

chore: replace legacy `jax.random.PRNGKey` with modern `jax.random.key` (#2134)

* chore: replace legacy `jax.random.PRNGKey` with modern `jax.random.key`

* fix: files reformatted with `ruff==0.15.0`

* test: update PRNG key test

* test: update test effected by the `optax==0.2.7` release

* Revert "test: update test effected by the `optax==0.2.7` release"

This reverts commit 14d3ee8ac.

* Revert "fix: files reformatted with `ruff==0.15.0`"

This reverts commit 24827812e.

* test: pickle `jax.random.key`

* test: use legacy keys in `test/test_pickle.py` until
https://github.com/jax-ml/jax/issues/35065 is resolved

6 of 6 new or added lines in 5 files covered. (100.0%)

14055 of 15179 relevant lines covered (92.6%)

1.69 hits per line