From 4457e37b39bfc660beb8be579b282e5acbae1b5f Mon Sep 17 00:00:00 2001 From: Aaron Meurer Date: Mon, 28 Mar 2022 20:32:39 -0600 Subject: Fix a type hint in numpy.array_api --- numpy/array_api/linalg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'numpy/array_api') 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 `. -- cgit v1.2.1