pyro-ppl / numpyro / 42107037275

93%

push

github

fix(distributions): promote discrete and gate logits before evaluation (#2397)

* fix(distributions): promote Bernoulli and gate logits before evaluation

* Promote sibling count and categorical logits and cover public predictive factories

* Handle traced categorical list logits with the shared promotion helper

* fix(distributions): promote multinomial logits and preserve weak Python scalars

Address the remaining sampling and dtype cases in review of #2397. Python int and bool parameters become Python floats, retaining JAX weak typing; NumPy and traced integer arrays still become strong floating arrays. Include Multinomial in factory, method, list, host-storage and Predictive coverage, with scalar dtype controls for eager and JIT paths. Restore the SciPy import used by newly merged upstream Bernoulli tests.

* test(distributions): trim redundant integer parameter cases

* test(distributions): keep the existing host storage rows

The trim replaced the five host storage cases from #2375 with the single
CategoricalLogits case; keep them and add CategoricalLogits to them.

11 of 11 new or added lines in 3 files covered. (100.0%)

15892 of 17101 relevant lines covered (92.93%)

1.7 hits per line