🎨 linalg: simplified _matfuncs type aliases

Author: jorenham

scipy-stubs/linalg/_matfuncs.pyi

@@ -41,6 +41,8 @@ _Complex128ND: TypeAlias = onp.ArrayND[np.complex128]
 _ComplexND: TypeAlias = onp.ArrayND[npc.complexfloating]
 _InexactND: TypeAlias = onp.ArrayND[npc.inexact]
 
+_AsFloat64ND: TypeAlias = onp.ToArrayND[float, np.float64 | npc.integer | np.bool_]
+
 ###
 
 eps: Final[np.float64] = ...  # undocumented
@@ -77,7 +79,7 @@ def fractional_matrix_power(A: onp.ToComplexND, t: onp.ToFloat) -> onp.ArrayND[A
 
 #
 @overload
-def sqrtm(A: onp.ToIntND | onp.ToJustFloat64_ND) -> _Float64ND: ...
+def sqrtm(A: _AsFloat64ND) -> _Float64ND: ...
 @overload
 def sqrtm(A: onp.ToFloatND) -> _FloatND: ...
 @overload
@@ -106,7 +108,7 @@ def sqrtm(A: onp.ToComplexND, disp: onp.ToFalse, blocksize: int) -> tuple[_Inexa
 
 # NOTE: return dtype depends on the sign of the values
 @overload  # +integer | ~float64
-def logm(A: onp.ToIntND | onp.ToJustFloat64_ND) -> _Float64ND | _Complex128ND: ...
+def logm(A: _AsFloat64ND) -> _Float64ND | _Complex128ND: ...
 @overload  # +floating
 def logm(A: onp.ToFloatND) -> _InexactND: ...
 @overload  # ~complex128
@@ -127,7 +129,7 @@ def logm(A: onp.ToComplexND, disp: onp.ToFalse) -> tuple[_InexactND, float]: ...
 
 #
 @overload  # +integer | ~float64
-def expm(A: onp.ToIntND | onp.ToJustFloat64_ND) -> _Float64ND: ...
+def expm(A: _AsFloat64ND) -> _Float64ND: ...
 @overload  # +floating
 def expm(A: onp.ToFloatND) -> _FloatND: ...
 @overload  # ~complex128
@@ -142,7 +144,7 @@ def _exp_sinch(x: onp.ArrayND[_ComplexT, _ShapeT]) -> onp.ArrayND[_ComplexT, _Sh
 
 #
 @overload  # +integer | ~float64
-def cosm(A: onp.ToIntND | onp.ToJustFloat64_ND) -> _Float64ND: ...
+def cosm(A: _AsFloat64ND) -> _Float64ND: ...
 @overload  # +floating
 def cosm(A: onp.ToFloatND) -> _FloatND: ...
 @overload  # ~complex128
@@ -154,7 +156,7 @@ def cosm(A: onp.ToComplexND) -> _InexactND: ...
 
 #
 @overload  # +integer | ~float64
-def sinm(A: onp.ToIntND | onp.ToJustFloat64_ND) -> _Float64ND: ...
+def sinm(A: _AsFloat64ND) -> _Float64ND: ...
 @overload  # +floating
 def sinm(A: onp.ToFloatND) -> _FloatND: ...
 @overload  # ~complex128
@@ -166,7 +168,7 @@ def sinm(A: onp.ToComplexND) -> _InexactND: ...
 
 #
 @overload  # +integer | ~float64
-def tanm(A: onp.ToIntND | onp.ToJustFloat64_ND) -> _Float64ND: ...
+def tanm(A: _AsFloat64ND) -> _Float64ND: ...
 @overload  # +floating
 def tanm(A: onp.ToFloatND) -> _FloatND: ...
 @overload  # ~complex128
@@ -178,7 +180,7 @@ def tanm(A: onp.ToComplexND) -> _InexactND: ...
 
 #
 @overload  # +integer | ~float64
-def coshm(A: onp.ToIntND | onp.ToJustFloat64_ND) -> _Float64ND: ...
+def coshm(A: _AsFloat64ND) -> _Float64ND: ...
 @overload  # +floating
 def coshm(A: onp.ToFloatND) -> _FloatND: ...
 @overload  # ~complex128
@@ -190,7 +192,7 @@ def coshm(A: onp.ToComplexND) -> _InexactND: ...
 
 #
 @overload  # +integer | ~float64
-def sinhm(A: onp.ToIntND | onp.ToJustFloat64_ND) -> _Float64ND: ...
+def sinhm(A: _AsFloat64ND) -> _Float64ND: ...
 @overload  # +floating
 def sinhm(A: onp.ToFloatND) -> _FloatND: ...
 @overload  # ~complex128
@@ -202,7 +204,7 @@ def sinhm(A: onp.ToComplexND) -> _InexactND: ...
 
 #
 @overload  # +integer | ~float64
-def tanhm(A: onp.ToIntND | onp.ToJustFloat64_ND) -> _Float64ND: ...
+def tanhm(A: _AsFloat64ND) -> _Float64ND: ...
 @overload  # +floating
 def tanhm(A: onp.ToFloatND) -> _FloatND: ...
 @overload  # ~complex128
@@ -228,7 +230,7 @@ def funm(A: onp.CanArrayND[_InexactT], func: _FuncND[_InexactT], disp: onp.ToFal
 
 #
 @overload  # +float64
-def signm(A: onp.ToIntND | onp.ToFloat64_ND) -> _Float64ND: ...
+def signm(A: _AsFloat64ND) -> _Float64ND: ...
 @overload  # +floating
 def signm(A: onp.ToFloatND) -> _FloatND: ...
 @overload  # +complexfloating
@@ -248,7 +250,7 @@ def signm(A: onp.ToComplexND, disp: onp.ToTrue) -> _InexactND: ...
 def signm(A: onp.ToComplexND, disp: onp.ToFalse) -> tuple[_InexactND, np.float64]: ...
 
 #
-@overload  # +integer, +integer
+@overload  # +integer | ~float64
 def khatri_rao(a: onp.ToIntND, b: onp.ToIntND) -> _IntND: ...
 @overload  # +float64, ~float64
 def khatri_rao(a: onp.ToFloat64_ND, b: onp.ToJustFloat64_ND) -> _Float64ND: ...