Merge pull request #117 from asmeurer/std-fix

Fix torch.std() when correction is an integer

Author: asmeurer

array_api_compat/torch/_aliases.py

@@ -384,6 +384,8 @@ def std(x: array,
         _correction = int(correction)
         if correction != _correction:
             raise NotImplementedError("float correction in torch std() is not yet supported")
+    else:
+        _correction = correction
 
     # https://github.com/pytorch/pytorch/issues/29137
     if axis == ():

numpy-1-21-xfails.txt

@@ -130,6 +130,7 @@ array_api_tests/test_operators_and_elementwise_functions.py::test_floor_divide[_
 array_api_tests/test_operators_and_elementwise_functions.py::test_floor_divide[floor_divide(x1, x2)]
 array_api_tests/test_operators_and_elementwise_functions.py::test_greater[__gt__(x1, x2)]
 array_api_tests/test_operators_and_elementwise_functions.py::test_greater[greater(x1, x2)]
+array_api_tests/test_operators_and_elementwise_functions.py::test_greater_equal[greater_equal(x1, x2)]
 array_api_tests/test_operators_and_elementwise_functions.py::test_less[__lt__(x1, x2)]
 array_api_tests/test_operators_and_elementwise_functions.py::test_less[less(x1, x2)]
 array_api_tests/test_operators_and_elementwise_functions.py::test_less_equal[less_equal(x1, x2)]