Skip to content

Commit 0fb2e32

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

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

‎scipy-stubs/stats/_stats_py.pyi‎

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ from typing_extensions import NamedTuple, TypeVar, deprecated
66

77
import numpy as np
88
import numpy.typing as npt
9+
import numpy_typing_compat as nptc
910
import optype as op
1011
import optype.numpy as onp
1112
import optype.numpy.compat as npc
@@ -689,14 +690,14 @@ def sem(
689690
# NOTE: keep in sync with `gzscore`
690691
@overload # +integer, known shape
691692
def zscore(
692-
a: onp.CanArray[_ShapeT, np.dtype[npc.integer | np.bool_]],
693+
a: nptc.CanArray[_ShapeT, np.dtype[npc.integer | np.bool_]],
693694
axis: int | None = 0,
694695
ddof: int = 0,
695696
nan_policy: NanPolicy = "propagate",
696697
) -> onp.ArrayND[np.float64, _ShapeT]: ...
697698
@overload # known inexact dtype, known shape
698699
def zscore(
699-
a: onp.CanArray[_ShapeT, np.dtype[_InexactT]], axis: int | None = 0, ddof: int = 0, nan_policy: NanPolicy = "propagate"
700+
a: nptc.CanArray[_ShapeT, np.dtype[_InexactT]], axis: int | None = 0, ddof: int = 0, nan_policy: NanPolicy = "propagate"
700701
) -> onp.ArrayND[_InexactT, _ShapeT]: ...
701702
@overload # float 1d
702703
def zscore(
@@ -734,15 +735,15 @@ def zscore(
734735
# NOTE: keep in sync with `zscore`
735736
@overload # +integer, known shape
736737
def gzscore(
737-
a: onp.CanArray[_ShapeT, np.dtype[npc.integer | np.bool_]],
738+
a: nptc.CanArray[_ShapeT, np.dtype[npc.integer | np.bool_]],
738739
*,
739740
axis: int | None = 0,
740741
ddof: int = 0,
741742
nan_policy: NanPolicy = "propagate",
742743
) -> onp.ArrayND[np.float64, _ShapeT]: ...
743744
@overload # known inexact dtype, known shape
744745
def gzscore(
745-
a: onp.CanArray[_ShapeT, np.dtype[_InexactT]], *, axis: int | None = 0, ddof: int = 0, nan_policy: NanPolicy = "propagate"
746+
a: nptc.CanArray[_ShapeT, np.dtype[_InexactT]], *, axis: int | None = 0, ddof: int = 0, nan_policy: NanPolicy = "propagate"
746747
) -> onp.ArrayND[_InexactT, _ShapeT]: ...
747748
@overload # float 1d
748749
def gzscore(

0 commit comments

Comments
 (0)