pyro-ppl / numpyro / 36903396962

93%

push

github

fix(distributions): handle the deterministic GeometricProbs endpoint (#2298)

* fix(distributions): handle the deterministic GeometricProbs endpoint

Co-authored-by: Codex <noreply@openai.com>

* Respect enabled JAX precision in Geometric endpoint tests

Co-authored-by: Codex <noreply@openai.com>

* test(distributions): focus geometric endpoint regression

Replace redundant test variants with the single endpoint regression requested by juanitorduz. Preserve the source fix and verify log probabilities, entropy, and the where-guard gradient.

Co-authored-by: Codex <noreply@openai.com>

---------

Co-authored-by: Codex <noreply@openai.com>

2 of 2 new or added lines in 1 file covered. (100.0%)

15643 of 16873 relevant lines covered (92.71%)

1.7 hits per line