pyro-ppl / numpyro / 39302710555

93%

push

github

fix(infer): pair batched model and guide parameters in Predictive (#2371)

I found that Predictive sliced guide parameters by particle but reused the full batch of model parameters, producing an extra axis and mixing parameter sets. I now substitute the current parameter slice into both the model and guide. I added a regression for both prediction modes and clarified the batched-parameter documentation.

Assisted-by: Claude
Assisted-by: GPT-6 Astra

Signed-off-by: mottopanikeiku <176798723+mottopanikeiku@users.noreply.github.com>
Co-authored-by: mottopanikeiku <176798723+mottopanikeiku@users.noreply.github.com>

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

15753 of 16977 relevant lines covered (92.79%)

1.7 hits per line