pyro-ppl / numpyro / 18640794209

92%

Build Type

push

github

Committed by web-flow

Commit Message

fix(gh-2036): MyPy Errors in `numpyro.distributions.transforms` Module (#2066)

* chore: incomplete changes for type hint in numpyro.distribution.transforms

* refactor: update type hints and improve type safety across transforms and constraints

* fix: improve type casting in Transform class and ensure proper handling of inverse transforms

Co-authored-by: Juan Orduz <juanitorduz@gmail.com>

* chore: remove `numpy.typing.mypy_plugin` from mypy configuration

Co-authored-by: Juan Orduz <juanitorduz@gmail.com>

* fix: update `_inv` type to support weak references in `Transform` class

Co-authored-by: Juan Orduz <juanitorduz@gmail.com>

* fix: update `intermediates` type to `Optional[PyTree]` in Transform classes and add PyTree type alias

* fix: update type hints in `Transform` classes to use `NonScalarArray` and `NumLike`

* fix: return type of `log_abs_det_jacobian` changed to `NumLike`

* fix: update `call_with_intermediates` method to return `Optional[PyTree]`

* using Generic type for the base transform with NumLike bound

* formatting transform methods

* fix typing of ComplexTransform

* format transforms module again

* fix for transform.inv might get value None

* revert to use jnp.array_equal to compare the domains

* fix lint

---------

Co-authored-by: Juan Orduz <juanitorduz@gmail.com>
Co-authored-by: Du Phan <fehiepsi@gmail.com>

Run Details

261 of 264 new or added lines in 4 files covered. (98.86%)

1 existing line in 1 file now uncovered.

13704 of 14822 relevant lines covered (92.46%)

1.69 hits per line