summaryrefslogtreecommitdiff
path: root/numpy/array_api/linalg.py
diff options
context:
space:
mode:
Diffstat (limited to 'numpy/array_api/linalg.py')
-rw-r--r--numpy/array_api/linalg.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/numpy/array_api/linalg.py b/numpy/array_api/linalg.py
index 8398147ba..f422e1c27 100644
--- a/numpy/array_api/linalg.py
+++ b/numpy/array_api/linalg.py
@@ -384,7 +384,7 @@ def vecdot(x1: Array, x2: Array, /, *, axis: int = -1) -> Array:
# The type for ord should be Optional[Union[int, float, Literal[np.inf,
# -np.inf]]] but Literal does not support floating-point literals.
-def vector_norm(x: Array, /, *, axis: Optional[Union[int, Tuple[int, int]]] = None, keepdims: bool = False, ord: Optional[Union[int, float]] = 2) -> Array:
+def vector_norm(x: Array, /, *, axis: Optional[Union[int, Tuple[int, ...]]] = None, keepdims: bool = False, ord: Optional[Union[int, float]] = 2) -> Array:
"""
Array API compatible wrapper for :py:func:`np.linalg.norm <numpy.linalg.norm>`.