Skip to content

Commit d4fcd8c

Browse files
committed
Fix a type hint in numpy.array_api
Original NumPy Commit: 4457e37b39bfc660beb8be579b282e5acbae1b5f
1 parent a2711e6 commit d4fcd8c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

array_api_strict/linalg.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -384,7 +384,7 @@ def vecdot(x1: Array, x2: Array, /, *, axis: int = -1) -> Array:
384384

385385
# The type for ord should be Optional[Union[int, float, Literal[np.inf,
386386
# -np.inf]]] but Literal does not support floating-point literals.
387-
def vector_norm(x: Array, /, *, axis: Optional[Union[int, Tuple[int, int]]] = None, keepdims: bool = False, ord: Optional[Union[int, float]] = 2) -> Array:
387+
def vector_norm(x: Array, /, *, axis: Optional[Union[int, Tuple[int, ...]]] = None, keepdims: bool = False, ord: Optional[Union[int, float]] = 2) -> Array:
388388
"""
389389
Array API compatible wrapper for :py:func:`np.linalg.norm <numpy.linalg.norm>`.
390390

0 commit comments

Comments
 (0)