✨ linalg: improved eigh return type inference for float64

Author: jorenham

scipy-stubs/linalg/_decomp.pyi

@@ -384,7 +384,22 @@ def eig(
     ...
 
 #
-@overload  # float, eigvals_only: False = ...
+@overload  # +float64, eigvals_only: False = ...
+def eigh(  #
+    a: onp.ToArrayND[float, np.float64 | np.longdouble | npc.integer64 | npc.integer32],
+    b: onp.ToFloat64_ND | None = None,
+    *,
+    lower: op.CanBool = True,
+    eigvals_only: onp.ToFalse = False,
+    overwrite_a: op.CanBool = False,
+    overwrite_b: op.CanBool = False,
+    type: _EigHType = 1,
+    check_finite: op.CanBool = True,
+    subset_by_index: _EigHSubsetByIndex | None = None,
+    subset_by_value: _EigHSubsetByValue | None = None,
+    driver: _DriverEV | _DriverGV | None = None,
+) -> tuple[onp.ArrayND[np.float64], onp.ArrayND[np.float64]]: ...
+@overload  # +float, eigvals_only: False = ...
 def eigh(
     a: onp.ToFloatND,
     b: onp.ToFloatND | None = None,
@@ -399,22 +414,22 @@ def eigh(
     subset_by_value: _EigHSubsetByValue | None = None,
     driver: _DriverEV | _DriverGV | None = None,
 ) -> tuple[_FloatND, _FloatND]: ...
-@overload  # float, eigvals_only: True
+@overload  # ~complex, eigvals_only: False = ...
 def eigh(
-    a: onp.ToFloatND,
-    b: onp.ToFloatND | None = None,
+    a: onp.ToJustComplexND,
+    b: onp.ToComplexND | None = None,
     *,
     lower: op.CanBool = True,
-    eigvals_only: onp.ToTrue,
+    eigvals_only: onp.ToFalse = False,
     overwrite_a: op.CanBool = False,
     overwrite_b: op.CanBool = False,
     type: _EigHType = 1,
     check_finite: op.CanBool = True,
     subset_by_index: _EigHSubsetByIndex | None = None,
     subset_by_value: _EigHSubsetByValue | None = None,
-    driver: _DriverEV | _EigHSubsetByValue | None = None,
-) -> _FloatND: ...
-@overload  # complex, eigvals_only: False = ...
+    driver: _DriverEV | _DriverGV | None = None,
+) -> tuple[_FloatND, _ComplexND]: ...
+@overload  # +complex, eigvals_only: False = ...
 def eigh(
     a: onp.ToComplexND,
     b: onp.ToComplexND | None = None,
@@ -429,10 +444,10 @@ def eigh(
     subset_by_value: _EigHSubsetByValue | None = None,
     driver: _DriverEV | _DriverGV | None = None,
 ) -> tuple[_FloatND, _InexactND]: ...
-@overload  # complex, eigvals_only: True
+@overload  # +complex128, eigvals_only: True
 def eigh(
-    a: onp.ToComplexND,
-    b: onp.ToComplexND | None = None,
+    a: onp.ToArrayND[float, npc.inexact80 | npc.number64 | npc.integer32],
+    b: onp.ToComplex128_ND | None = None,
     *,
     lower: op.CanBool = True,
     eigvals_only: onp.ToTrue,
@@ -443,22 +458,22 @@ def eigh(
     subset_by_index: _EigHSubsetByIndex | None = None,
     subset_by_value: _EigHSubsetByValue | None = None,
     driver: _DriverEV | _EigHSubsetByValue | None = None,
-) -> _FloatND: ...
-@overload  # complex, eigvals_only: CanBool (catch-all)
+) -> onp.ArrayND[np.float64]: ...
+@overload  # +complex, eigvals_only: True
 def eigh(
     a: onp.ToComplexND,
     b: onp.ToComplexND | None = None,
     *,
-    lower: op.CanBool,
-    eigvals_only: op.CanBool,
+    lower: op.CanBool = True,
+    eigvals_only: onp.ToTrue,
     overwrite_a: op.CanBool = False,
     overwrite_b: op.CanBool = False,
     type: _EigHType = 1,
     check_finite: op.CanBool = True,
     subset_by_index: _EigHSubsetByIndex | None = None,
     subset_by_value: _EigHSubsetByValue | None = None,
     driver: _DriverEV | _EigHSubsetByValue | None = None,
-) -> _FloatND | tuple[_FloatND, _InexactND]: ...
+) -> _FloatND: ...
 
 #
 @overload  # float, eigvals_only: False = ..., select: _SelectA = ...