XENONnT / straxen / 10552369160

89%

push

github

Plugins for position reconstruction with conditional normalizing flow (#1404)

* Add normalizing flow position reconstruction plugin, and modify existing peak position plugin to include the flow plugin.
Committer: Juehang Qin <qinjuehang@rice.edu>
On branch flow-posrec
Changes to be committed:
	modified:   straxen/plugins/peaks/peak_positions.py
	new file:   straxen/plugins/peaks/peak_positions_flow.py

* Added URLConfig protocol for serialized jax functions

* Add flow plugin to peaks __init__.py

* Add flow Event posrec plugins

* Add flow Event posrec plugins to __init__.py

* Restore default_reconstruction_algorithm in peak_positions

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Remove default_reconstruction_algorithm from EventPosEventPositionUncertainty where it is not used

* fix merge typo

* Add numpy import to peak_positions_flow.py

* Adds check to make sure model file exists in tar.gz resource

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Minor fix to make sure mypy correctly infers type in jax URLConfig

* another mypy fix

* flake8 style fixes

* fix flow plugin default model URLConfig

* Update documentation for flow plugins

* Added note to docstrings, changed flow to cnf, removed extra var

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Change flow to cnf in peak_positions plugin

* Add no-qa for flake8 line length check in docstring URL.

* Add default_reconstruction_algorithm

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* `PeakPositionsCNF` should be inherited from `PeakPositionsBaseNT`

---------

Co-authored-by: Juehang Qin <juehang@midway3-login4.rcc.local>
Co-authored-by: Juehang Qin <qinjuehang@rice.edu>
Co-authored-by: pre... (continued)

170 of 190 new or added lines in 6 files covered. (89.47%)

9157 of 10084 relevant lines covered (90.81%)

1.82 hits per line