@@ -689,14 +689,11 @@ def sem(
689689# NOTE: keep in sync with `gzscore`
690690@overload # +integer, known shape
691691def 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
698695def 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
702699def zscore (
@@ -734,15 +731,15 @@ def zscore(
734731# NOTE: keep in sync with `zscore`
735732@overload # +integer, known shape
736733def 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
744741def 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
748745def gzscore (
0 commit comments