pyro-ppl / numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
https://num.pyro.ai
Apache License 2.0
2.1k stars 227 forks source link

Add a warning message when using prng_key outside of seed #1760

Closed fehiepsi closed 3 months ago

fehiepsi commented 4 months ago

Fixes #1741

martinjankowiak commented 3 months ago

yikes looks like merging these old PRs caused linting issues

ruff check .
All checks passed!
ruff format . --check
Would reformat: numpyro/infer/reparam.py
1 file would be reformatted, 1[8](https://github.com/pyro-ppl/numpyro/actions/runs/8402793380/job/23012559867#step:5:9)[9](https://github.com/pyro-ppl/numpyro/actions/runs/8402793380/job/23012559867#step:5:10) files already formatted
make: *** [Makefile:5: lint] Error 1
Error: Process completed with exit code 2.