chore: update numpyro requirement from <0.22,>=0.19.0 to >=0.19.0,<0.23

[dependabot]

Updates the requirements on numpyro to permit the latest version.

Release notes

Sourced from numpyro's releases.

0.22.0

What's Changed

• Fix effect handlers SWIG reference by @​jsakv in pyro-ppl/numpyro#2184
• Fix incorrect HPDI docstring by @​ordabayevy in pyro-ppl/numpyro#2183
• fix(gh-2181): ensure consistent behavior of Poisson.log_prob by casting rate to floating-point type by @​Qazalbash in pyro-ppl/numpyro#2182
• doc: detailed documentation of Beta and BetaProportion distribution by @​Qazalbash in pyro-ppl/numpyro#2185
• doc(gh-2187): add math explanations for Normal, Cauchy, and Exponential distributions by @​kimjune01 in pyro-ppl/numpyro#2188
• dev: pre-commit hook to standardize toml format by @​Qazalbash in pyro-ppl/numpyro#2186
• fix: freshly formatted toml files by @​Qazalbash in pyro-ppl/numpyro#2190
• doc: replace XXX with Note in the code comments by @​star1327p in pyro-ppl/numpyro#2191
• Fix incorrect statement in GammaPoisson log_prob docstring by @​dylanhmorris in pyro-ppl/numpyro#2194
• Expose NegativeBinomial from top-level distributions module by @​dylanhmorris in pyro-ppl/numpyro#2196
• Expose ZeroInflatedLogits and ZeroInflatedProbs from top-level distributions module by @​Qazalbash in pyro-ppl/numpyro#2197
• Fix EnsembleSampler second split to use complementary inactive chains by @​amifalk in pyro-ppl/numpyro#2189
• doc(gh-2187): enhance documentation for Bernoulli and Binomial distributions by @​Qazalbash in pyro-ppl/numpyro#2192
• Fix path Boston housing dataset by @​juanitorduz in pyro-ppl/numpyro#2200
• doc(gh-2187): enhance documentation for Chi2, DiscreteUniform, and Categorical distributions by @​Metbcy in pyro-ppl/numpyro#2199
• doc(gh-2187): docstrings for Gumbel, Laplace, Logistic by @​Metbcy in pyro-ppl/numpyro#2202
• Hurdle Distributions by @​juanitorduz in pyro-ppl/numpyro#2195
• Fix enable_validation by @​renecotyfanboy in pyro-ppl/numpyro#2201
• Fix GaussianStateSpace.variance shape when batch comes from initial_value by @​tillahoffmann in pyro-ppl/numpyro#2205
• Migrate from MyPy to Ty by @​juanitorduz in pyro-ppl/numpyro#2207
• Add HaarTransform and DiscreteCosineTransform by @​juanitorduz in pyro-ppl/numpyro#2208
• Better casting eps by @​juanitorduz in pyro-ppl/numpyro#2209
• Improve integration with external samplers by @​juanitorduz in pyro-ppl/numpyro#2203
• Rename var -> variance for truncated distributions and close support. by @​tillahoffmann in pyro-ppl/numpyro#2210
• Improve type hints for the HSGP contrib module by @​juanitorduz in pyro-ppl/numpyro#2211
• docs: document NUTS regularize_mass_matrix by @​ahfoysal in pyro-ppl/numpyro#2214
• Pin ty and improve type hints by @​juanitorduz in pyro-ppl/numpyro#2218
• Miscellaneous bugfixes by @​esennesh in pyro-ppl/numpyro#2215
• Avoid unnecessary deepcopy in random_*_module and promote_batch_shape by @​kyo219 in pyro-ppl/numpyro#2222
• Fix stale trace count in test_mcmc_parallel_chain and run it in CI by @​esennesh in pyro-ppl/numpyro#2223
• chore: format README with recent ruff==0.16.0 by @​Qazalbash in pyro-ppl/numpyro#2228
• doc(gh-2187): docstrings for Dirichlet and Geometric by @​kir-7 in pyro-ppl/numpyro#2227
• doc: update README for header structure by @​Qazalbash in pyro-ppl/numpyro#2231
• doc: fix stale instructions in docs/README by @​Qazalbash in pyro-ppl/numpyro#2232
• feat: compare branch speed against master and report it on the PR by @​Qazalbash in pyro-ppl/numpyro#2233
• ci: add PR triage workflow for auto review requests and labels by @​Qazalbash in pyro-ppl/numpyro#2234
• ci: use pull_request_target for PR triage workflow by @​Qazalbash in pyro-ppl/numpyro#2237
• chore: workflow to clean pr after merging by @​Qazalbash in pyro-ppl/numpyro#2239
• Avoid creating NaNs in binary_cross_entropy_with_logits by @​miguelbiron in pyro-ppl/numpyro#2241
• ci: grant issues:write to pr-cleanup workflow by @​Qazalbash in pyro-ppl/numpyro#2242
• Validate num_samples in MCMC initialization by @​haansn08 in pyro-ppl/numpyro#2236
• fix NaN mixture log_prob gradient at zero weights by @​esennesh in pyro-ppl/numpyro#2224
• Fix provenance tracking for closed-over constants on JAX 0.11.1 by @​tandede in pyro-ppl/numpyro#2245
• Speed up posterior diagnostics (summary 1.7x, effective_sample_size 2.1x) by @​kyo219 in pyro-ppl/numpyro#2251
• Drop the masked logsumexp reduction in mixture log_prob by @​kyo219 in pyro-ppl/numpyro#2250
• Fix generator being appended instead of a list in scan carry_shapes by @​kyo219 in pyro-ppl/numpyro#2248
• Avoid per-step host sync in the SVI progress-bar loop by @​kyo219 in pyro-ppl/numpyro#2247
• Add cat constraint and CatTransform by @​aswanth-07 in pyro-ppl/numpyro#2246

... (truncated)

Commits

• 1e5c0bf run script (#2275)
• 0771b92 fix(distributions): keep Categorical entropy finite when a category has zero ...
• 0adc509 Make Gibbs kernels composable (#2249)
• 0830c6d update censoring notebook (#2235)
• 07501df doc: link citation markers in the contrib API docstrings (#2264)
• 9e26524 doc(gh-2187): mathematical details for StudentT (#2268)
• 1277555 refactor: keep truncation bounds out of the static treedef for the power-law ...
• b1a08d4 doc(gh-2187): mathematical details for VonMises (#2270)
• 7012982 doc(gh-2187): mathematical details for InverseGamma and Kumaraswamy (#2269)
• 5623d64 docs(gh-2187): add mathematical details for Pareto and Weibull (#2265)
• Additional commits viewable in compare view

[coderabbitai]

Important

Review skipped

Bot user detected.

To trigger a single review, invoke the @coderabbitai review command.

⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Advanced

Run ID: f5bdd744-ae1b-4aee-8bf1-18d9de34f27b

You can disable this status message by setting the reviews.review_status to false in the CodeRabbit configuration file.

Use the checkbox below for a quick retry:

• 🔍 Trigger review

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands.}

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 82.15%. Comparing base (b64503c) to head (f3ee9c7).
✅ All tests successful. No failed tests found.

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #352      +/-   ##
==========================================
+ Coverage   82.14%   82.15%   +0.01%     
==========================================
  Files          44       44              
  Lines        9140     9140              
==========================================
+ Hits         7508     7509       +1     
+ Misses       1632     1631       -1     

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.