Skip to content

Commit 35b06e5

Browse files
committed
1 parent f486969 commit 35b06e5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

array_api_compat/torch/_aliases.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -694,7 +694,7 @@ def triu(x: Array, /, *, k: int = 0) -> Array:
694694
return torch.triu(x, k)
695695

696696
# Functions that aren't in torch https://github.com/pytorch/pytorch/issues/58742
697-
def expand_dims(x: Array, /, *, axis: int | tuple[int, ...]) -> Array:
697+
def expand_dims(x: Array, /, axis: int | tuple[int, ...]) -> Array:
698698
if isinstance(axis, int):
699699
return torch.unsqueeze(x, axis)
700700
else:

0 commit comments

Comments
 (0)