complete scipy.linalg.norm

jorenham · Oct 18, 2024 · c3251f4 · c3251f4
1 parent 19cbb82
commit c3251f4
Showing 1 changed file with 21 additions and 3 deletions.
diff --git a/scipy-stubs/linalg/_misc.pyi b/scipy-stubs/linalg/_misc.pyi
@@ -1,18 +1,36 @@
-from typing import Literal
+from typing import Literal, overload
 
 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 Untyped
 
 __all__ = ["LinAlgError", "LinAlgWarning", "norm"]
 
 class LinAlgWarning(RuntimeWarning): ...
 
+@overload
 def norm(
     a: npt.ArrayLike,
     ord: Literal["fro", "nuc", 0, 1, -1, 2, -2] | float | None = None,
-    axis: Untyped | None = None,
+    axis: None = None,
+    keepdims: bool = False,
+    check_finite: bool = 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,
+) -> 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,
 ) -> np.float64 | npt.NDArray[np.float64]: ...