pyro-ppl / numpyro / 21185503949

93%

push

github

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

* feat: enhance types in `numpyro.distributions.constraints` module

* refactor: simplify `ConstraintT` protocol by removing generic type parameter

* refactor: remove unused `NumLikeT` type variable and streamline imports in typing modules

* refactor: remove setter methods for `is_discrete` and `event_dim` in `Constraint` class

* refactor: replace `reshape` method with `jnp.reshape` in `_IndependentConstraint` class

* refactor: replace `jnp` with dynamic use of `jax.numpy` or `numpy`

* Revert "refactor: remove unused `NumLikeT` type variable and streamline imports in typing modules"

This reverts commit 78d8e93af.

* refactor: replace `jnp` with `xp` and replace logical operations with bitwise operators

* refactor: update type parameters in `_RealVector` and `_RealMatrix` classes to use `NonScalarArray`

* refactor: replace private attributes with public properties for `is_discrete` and `event_dim` in `Constraint` class

* refactor: update type hints for constraints to use Union

* refactor: optimize logical check in `_PositiveOrderedVector` to use `jnp.all`

* fix: correction in attributes

* remove dependency on ConstraintT and TransformT

* Fix typing issues in handlers and primitives

* fix typing issues in IndependentTransform

---------

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

233 of 242 new or added lines in 5 files covered. (96.28%)

13996 of 15112 relevant lines covered (92.62%)

1.69 hits per line