Merge pull request #375 from ev-br/searchsorted_skip_dask

BUG: torch: expand_dims axis is keyword or positional argument

Author: ev-br

array_api_compat/torch/_aliases.py

@@ -694,7 +694,7 @@ def triu(x: Array, /, *, k: int = 0) -> Array:
     return torch.triu(x, k)
 
 # Functions that aren't in torch https://github.com/pytorch/pytorch/issues/58742
-def expand_dims(x: Array, /, *, axis: int | tuple[int, ...]) -> Array:
+def expand_dims(x: Array, /, axis: int | tuple[int, ...]) -> Array:
     if isinstance(axis, int):
         return torch.unsqueeze(x, axis)
     else:

dask-xfails.txt

@@ -135,10 +135,13 @@ array_api_tests/test_signatures.py::test_func_signature[broadcast_shapes]
 array_api_tests/test_data_type_functions.py::TestBroadcastShapes::test_broadcast_shapes
 array_api_tests/test_data_type_functions.py::TestBroadcastShapes::test_empty
 array_api_tests/test_data_type_functions.py::TestBroadcastShapes::test_error
+array_api_tests/test_searching_functions.py::test_searchsorted_with_scalars
 
 array_api_tests/test_linalg.py::test_eig
 array_api_tests/test_linalg.py::test_eigvals
 
+array_api_tests/test_manipulation_functions.py::TestExpandDims::test_expand_dims_tuples
+
 # Stubs have a comment: (**note**: libraries may return ``NaN`` to match Python behavior.)
 array_api_tests/test_special_cases.py::test_binary[floor_divide(x1_i is +infinity and isfinite(x2_i) and x2_i > 0) -> +infinity]
 array_api_tests/test_special_cases.py::test_binary[floor_divide(x1_i is +infinity and isfinite(x2_i) and x2_i < 0) -> -infinity]

numpy-1-26-xfails.txt

@@ -50,6 +50,8 @@ array_api_tests/test_operators_and_elementwise_functions.py::test_where_with_sca
 array_api_tests/test_creation_functions.py::test_meshgrid
 array_api_tests/test_data_type_functions.py::test_broadcast_arrays
 
+# observed with numpy==1.26 only, looks like is fixed on numpy 2.x
+array_api_tests/test_set_functions.py::TestIsin::test_isin_scalars
 
 # Stubs have a comment: (**note**: libraries may return ``NaN`` to match Python behavior.); Apparently, NumPy does just that
 array_api_tests/test_special_cases.py::test_binary[floor_divide(x1_i is +infinity and isfinite(x2_i) and x2_i > 0) -> +infinity]