pyro-ppl / numpyro / 31729612624

92%

push

github

fix NaN mixture log_prob gradient at zero weights (#2224)

* pyproject.toml: add pre-commit as a dev dependency like numpyro needs

Signed-off-by: Eli Sennesh <elisennesh@astera.org>

* numpyro.distributions.discrete: fix NaN log_prob gradients at zero weights

Testing Done: ???

Signed-off-by: Eli Sennesh <elisennesh@astera.org>

* distributions.mixtures.MixtureSameFamily: unwrap Independent constraints

Testing Done: pre-commit formatting checks pass, as does modified test

Signed-off-by: Eli Sennesh <elisennesh@gmail.com>

---------

Signed-off-by: Eli Sennesh <elisennesh@astera.org>
Signed-off-by: Eli Sennesh <elisennesh@gmail.com>

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

14619 of 15829 relevant lines covered (92.36%)

1.69 hits per line