Skip to content

Commit 69c405e

Browse files
committed
Fix torch.std() when correction is an integer
Fixes #116.
1 parent 94e42df commit 69c405e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

array_api_compat/torch/_aliases.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -384,6 +384,8 @@ def std(x: array,
384384
_correction = int(correction)
385385
if correction != _correction:
386386
raise NotImplementedError("float correction in torch std() is not yet supported")
387+
else:
388+
_correction = correction
387389

388390
# https://github.com/pytorch/pytorch/issues/29137
389391
if axis == ():

0 commit comments

Comments
 (0)