Merge pull request #21469 from jakevdp:array-api-update

PiperOrigin-RevId: 638328475

Author: jax authors

.github/workflows/jax-array-api.yml

@@ -25,7 +25,7 @@ jobs:
       with:
         repository: data-apis/array-api-tests
         # TODO(jakevdp) update this to a stable release/tag when available.
-        ref: 'e38ce3466e596ed2b8fa4638f161f5563ded81a8'  # Latest commit as of 2024-04-15
+        ref: '33f2d2ea2f3dd2b3ceeeb4519d55e08096184149'  # Latest commit as of 2024-05-28
         submodules: 'true'
         path: 'array-api-tests'
     - name: Set up Python ${{ matrix.python-version }}

jax/experimental/array_api/skips.txt

@@ -9,16 +9,27 @@ array_api_tests/test_array_object.py::test_setitem
 array_api_tests/test_special_cases.py::test_binary
 array_api_tests/test_special_cases.py::test_unary
 
-# fft test suite is buggy as of 83f0bcdc
-array_api_tests/test_fft.py
-
 # Pending implementation update for proper dtype promotion behavior,
 # see https://github.com/data-apis/array-api-tests/issues/234
 array_api_tests/test_statistical_functions.py::test_sum
 array_api_tests/test_statistical_functions.py::test_prod
-array_api_tests/test_linalg.py::test_trace
 
 # Pending bugfix, see https://github.com/data-apis/array-api-tests/issues/256
 array_api_tests/test_signatures.py::test_func_signature[logical_and]
 array_api_tests/test_signatures.py::test_func_signature[logical_or]
-array_api_tests/test_signatures.py::test_func_signature[logical_xor]
+array_api_tests/test_signatures.py::test_func_signature[logical_xor]
+
+# Returns int32 when int64 is expected
+array_api_tests/test_searching_functions.py::test_searchsorted
+
+# Various info functions not yet defined
+array_api_tests/test_has_names.py::test_has_names[info-capabilities]
+array_api_tests/test_has_names.py::test_has_names[info-default_device]
+array_api_tests/test_has_names.py::test_has_names[info-default_dtypes]
+array_api_tests/test_has_names.py::test_has_names[info-devices]
+array_api_tests/test_has_names.py::test_has_names[info-dtypes]
+array_api_tests/test_signatures.py::test_func_signature[capabilities]
+array_api_tests/test_signatures.py::test_func_signature[default_device]
+array_api_tests/test_signatures.py::test_func_signature[default_dtypes]
+array_api_tests/test_signatures.py::test_func_signature[devices]
+array_api_tests/test_signatures.py::test_func_signature[dtypes]