Skip to content

Commit 9eb938a

Browse files
TYP: Tighten the ord param of matrix_norm
Co-authored-by: Lucas Colley <[email protected]>
1 parent 920cb80 commit 9eb938a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: array_api_compat/common/_linalg.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def matrix_norm(
139139
xp: Namespace,
140140
*,
141141
keepdims: bool = False,
142-
ord: JustInt | JustFloat | Literal["fro", "nuc"] | None = "fro",
142+
ord: Literal[1, 2, -1, -2] | JustFloat | Literal["fro", "nuc"] | None = "fro",
143143
) -> Array:
144144
return xp.linalg.norm(x, axis=(-2, -1), keepdims=keepdims, ord=ord)
145145

0 commit comments

Comments
 (0)