pyro-ppl / numpyro / 12826727017

92%

push

github

Make `Delta.log_prob` jit-able on metal and raise explicit error for `Delta` sampling site during initialization. (#1950)

* Use `jnp.where` for `Delta.log_prob` (cf. jax-ml/jax#25935).

* Raise explicit error for unobserved `Delta` sample sites during intialization.

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

13296 of 14325 relevant lines covered (92.82%)

1.97 hits per line