✨ linalg: improved det annotations

Author: jorenham

scipy-stubs/linalg/_basic.pyi

@@ -1037,18 +1037,36 @@ def inv(
 ) -> onp.ArrayND[np.complex128, _ShapeT]: ...
 
 # TODO(jorenham): improve this
-@overload  # floating 2d
-def det(a: onp.ToFloatStrict2D, overwrite_a: bool = False, check_finite: bool = True) -> _Float: ...
-@overload  # floating 3d
-def det(a: onp.ToFloatStrict3D, overwrite_a: bool = False, check_finite: bool = True) -> _Float1D: ...
-@overload  # floating
-def det(a: onp.ToFloatND, overwrite_a: bool = False, check_finite: bool = True) -> _Float | _FloatND: ...
-@overload  # complexfloating 2d
-def det(a: onp.ToJustComplexStrict2D, overwrite_a: bool = False, check_finite: bool = True) -> _Inexact1D: ...
-@overload  # complexfloating 3d
-def det(a: onp.ToJustComplexStrict3D, overwrite_a: bool = False, check_finite: bool = True) -> _InexactND: ...
-@overload  # complexfloating
-def det(a: onp.ToComplexND, overwrite_a: bool = False, check_finite: bool = True) -> _Inexact | _InexactND: ...
+@overload  # +float64 2d
+def det(a: onp.ToFloat64Strict2D, overwrite_a: bool = False, check_finite: bool = True) -> np.float64: ...
+@overload  # +float64 3d
+def det(a: onp.ToFloat64Strict3D, overwrite_a: bool = False, check_finite: bool = True) -> onp.Array1D[np.float64]: ...
+@overload  # +float64 ND
+def det(a: onp.ToFloat64_ND, overwrite_a: bool = False, check_finite: bool = True) -> np.float64 | onp.ArrayND[np.float64]: ...
+@overload  # complex128 | complex64 2d
+def det(
+    a: onp.ToJustComplex128Strict2D | onp.CanArray2D[np.complex64], overwrite_a: bool = False, check_finite: bool = True
+) -> np.complex128: ...
+@overload  # complex128 | complex64 3d
+def det(
+    a: onp.ToJustComplex128Strict3D | onp.CanArray3D[np.complex64], overwrite_a: bool = False, check_finite: bool = True
+) -> onp.Array1D[np.complex128]: ...
+@overload  # complex128 | complex64 Nd
+def det(
+    a: onp.ToJustComplex128_ND, overwrite_a: bool = False, check_finite: bool = True
+) -> np.complex128 | onp.ArrayND[np.complex128]: ...
+@overload  # +complex128 2d
+def det(
+    a: onp.ToComplex128Strict2D | onp.CanArray2D[np.complex64], overwrite_a: bool = False, check_finite: bool = True
+) -> np.float64 | np.complex128: ...
+@overload  # +complex128 3d
+def det(
+    a: onp.ToComplex128Strict3D | onp.CanArray3D[np.complex64], overwrite_a: bool = False, check_finite: bool = True
+) -> onp.Array1D[np.float64 | np.complex128]: ...
+@overload  # +complex128 Nd
+def det(
+    a: onp.ToComplex128_ND, overwrite_a: bool = False, check_finite: bool = True
+) -> np.float64 | np.complex128 | onp.ArrayND[np.float64 | np.complex128]: ...
 
 # TODO(jorenham): improve this
 @overload  # (float[:, :], float[:]) -> (float[:], float[], ...)