Skip to content

Commit 9a56255

Browse files
committed
🩹 stats: fix [g]zscore shape-typing on numpy<2.1
1 parent 3ac1b7a commit 9a56255

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

‎scipy-stubs/stats/_stats_py.pyi‎

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -689,14 +689,11 @@ def sem(
689689
# NOTE: keep in sync with `gzscore`
690690
@overload # +integer, known shape
691691
def zscore(
692-
a: onp.CanArray[_ShapeT, np.dtype[npc.integer | np.bool_]],
693-
axis: int | None = 0,
694-
ddof: int = 0,
695-
nan_policy: NanPolicy = "propagate",
692+
a: onp.CanArrayND[npc.integer | np.bool_, _ShapeT], axis: int | None = 0, ddof: int = 0, nan_policy: NanPolicy = "propagate"
696693
) -> onp.ArrayND[np.float64, _ShapeT]: ...
697694
@overload # known inexact dtype, known shape
698695
def zscore(
699-
a: onp.CanArray[_ShapeT, np.dtype[_InexactT]], axis: int | None = 0, ddof: int = 0, nan_policy: NanPolicy = "propagate"
696+
a: onp.CanArrayND[_InexactT, _ShapeT], axis: int | None = 0, ddof: int = 0, nan_policy: NanPolicy = "propagate"
700697
) -> onp.ArrayND[_InexactT, _ShapeT]: ...
701698
@overload # float 1d
702699
def zscore(
@@ -734,15 +731,15 @@ def zscore(
734731
# NOTE: keep in sync with `zscore`
735732
@overload # +integer, known shape
736733
def gzscore(
737-
a: onp.CanArray[_ShapeT, np.dtype[npc.integer | np.bool_]],
734+
a: onp.CanArrayND[npc.integer | np.bool_, _ShapeT],
738735
*,
739736
axis: int | None = 0,
740737
ddof: int = 0,
741738
nan_policy: NanPolicy = "propagate",
742739
) -> onp.ArrayND[np.float64, _ShapeT]: ...
743740
@overload # known inexact dtype, known shape
744741
def gzscore(
745-
a: onp.CanArray[_ShapeT, np.dtype[_InexactT]], *, axis: int | None = 0, ddof: int = 0, nan_policy: NanPolicy = "propagate"
742+
a: onp.CanArrayND[_InexactT, _ShapeT], *, axis: int | None = 0, ddof: int = 0, nan_policy: NanPolicy = "propagate"
746743
) -> onp.ArrayND[_InexactT, _ShapeT]: ...
747744
@overload # float 1d
748745
def gzscore(

0 commit comments

Comments
 (0)