@@ -6,6 +6,7 @@ from typing_extensions import NamedTuple, TypeVar, deprecated
66
77import numpy as np
88import numpy .typing as npt
9+ import numpy_typing_compat as nptc
910import optype as op
1011import optype .numpy as onp
1112import optype .numpy .compat as npc
@@ -689,14 +690,14 @@ def sem(
689690# NOTE: keep in sync with `gzscore`
690691@overload # +integer, known shape
691692def 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
698699def 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
702703def zscore (
@@ -734,15 +735,15 @@ def zscore(
734735# NOTE: keep in sync with `zscore`
735736@overload # +integer, known shape
736737def 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
744745def 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
748749def gzscore (
0 commit comments