accept broader boolean arguments in scipy.linalg.norm

jorenham · Oct 18, 2024 · d5b3910 · d5b3910
1 parent ed07115
commit d5b3910
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/scipy-stubs/linalg/_misc.pyi b/scipy-stubs/linalg/_misc.pyi
@@ -4,6 +4,7 @@ import numpy as np
 import numpy.typing as npt
 import optype.typing as opt
 from numpy.linalg import LinAlgError  # noqa: ICN003
+from scipy._typing import AnyBool
 
 __all__ = ["LinAlgError", "LinAlgWarning", "norm"]
 
@@ -14,23 +15,23 @@ def norm(
     a: npt.ArrayLike,
     ord: Literal["fro", "nuc", 0, 1, -1, 2, -2] | float | None = None,
     axis: None = None,
-    keepdims: bool = False,
-    check_finite: bool = True,
+    keepdims: AnyBool = False,
+    check_finite: AnyBool = True,
 ) -> np.float64: ...
 @overload
 def norm(
     a: npt.ArrayLike,
     ord: Literal["fro", "nuc", 0, 1, -1, 2, -2] | float | None,
     axis: opt.AnyInt | tuple[opt.AnyInt, ...],
-    keepdims: bool = False,
-    check_finite: bool = True,
+    keepdims: AnyBool = False,
+    check_finite: AnyBool = True,
 ) -> np.float64 | npt.NDArray[np.float64]: ...
 @overload
 def norm(
     a: npt.ArrayLike,
     ord: Literal["fro", "nuc", 0, 1, -1, 2, -2] | float | None = None,
     *,
     axis: opt.AnyInt | tuple[opt.AnyInt, ...],
-    keepdims: bool = False,
-    check_finite: bool = True,
+    keepdims: AnyBool = False,
+    check_finite: AnyBool = True,
 ) -> np.float64 | npt.NDArray[np.float64]: ...