Skip to content

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

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 10 commits into
base: master
Choose a base branch
from
Open

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

Show file tree
Hide file tree
Changes from 6 commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
c12b514
feat: enhance types in `numpyro.distributions.constraints` module
Qazalbash Oct 21, 2025
cb01e09
refactor: simplify `ConstraintT` protocol by removing generic type pa…
Qazalbash Oct 21, 2025
78d8e93
refactor: remove unused `NumLikeT` type variable and streamline impor…
Qazalbash Oct 23, 2025
cd2bc8f
refactor: remove setter methods for `is_discrete` and `event_dim` in …
Qazalbash Oct 23, 2025
a6e52e1
refactor: replace `reshape` method with `jnp.reshape` in `_Independen…
Qazalbash Oct 23, 2025
5c331a1
refactor: replace `jnp` with dynamic use of `jax.numpy` or `numpy`
Qazalbash Oct 23, 2025
642ce7e
Revert "refactor: remove unused `NumLikeT` type variable and streamli…
Qazalbash Oct 24, 2025
def4e0b
refactor: replace `jnp` with `xp` and replace logical operations with…
Qazalbash Oct 24, 2025
b4e2b81
refactor: update type parameters in `_RealVector` and `_RealMatrix` c…
Qazalbash Oct 24, 2025
d609a7f
refactor: replace private attributes with public properties for `is_d…
Qazalbash Oct 24, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 4 additions & 3 deletions numpyro/_typing.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
except ImportError:
from typing_extensions import ParamSpec, TypeAlias


import numpy as np

import jax
Expand Down Expand Up @@ -45,10 +46,10 @@ def is_discrete(self) -> bool: ...
@property
def event_dim(self) -> int: ...

def __call__(self, x: ArrayLike) -> ArrayLike: ...
def __call__(self, x: NumLike) -> ArrayLike: ...
def __repr__(self) -> str: ...
def check(self, value: ArrayLike) -> ArrayLike: ...
def feasible_like(self, prototype: ArrayLike) -> ArrayLike: ...
def check(self, value: NumLike) -> ArrayLike: ...
def feasible_like(self, prototype: NumLike) -> NumLike: ...


@runtime_checkable
Expand Down
Loading
Loading