pyro-ppl / numpyro / 28484831643

92%

push

github

Improve integration with external samplers (#2203)

* init

* improvements

* update flowmc

* improvements

* type hints

* feedback 1

* missing fix

* cleanup

* docs

* rerun nb

* simplify

* rerun nb

* Address review: remove get_log_density_fn, inline via initialize_model

Per fehiepsi's review on PR #2203, drop the get_log_density_fn helper, its
LogDensityInfo return type, and the PositionDict alias; build the log
density inline by negating model_info.potential_fn (initialize_model
already returns a single-position potential_fn/postprocess_fn).

- numpyro/infer/util.py: remove get_log_density_fn + LogDensityInfo; narrow
  w.message to Warning in find_valid_initial_params so util.py type-checks
  under ty (newly added to [tool.ty.src]).
- numpyro/_typing.py, numpyro/infer/__init__.py, docs/source/utilities.rst:
  drop the removed symbols.
- notebooks/source/other_samplers.ipynb: inline the pattern and re-execute.
- test/infer/test_external_helpers.py: inline the pattern; keep constrain_fn
  and end-to-end MCMCKernel coverage.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

* rerun nb

* docs: clarify log-density wording in notebook

Reword the markdown lead-in to the Pathfinder log-density cell to say we
build the log-density function by negating the potential energy, avoiding
the awkward doubled 'negated' phrasing.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

* prune

---------

Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>

21 of 23 new or added lines in 2 files covered. (91.3%)

14571 of 15811 relevant lines covered (92.16%)

1.68 hits per line