diff --git a/.github/workflows/jax-array-api.yml b/.github/workflows/jax-array-api.yml index 01942e524ba5..3a1dd863c2b0 100644 --- a/.github/workflows/jax-array-api.yml +++ b/.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 }} diff --git a/jax/experimental/array_api/skips.txt b/jax/experimental/array_api/skips.txt index 5eb5fd8e27a5..49482b05fdf9 100644 --- a/jax/experimental/array_api/skips.txt +++ b/jax/experimental/array_api/skips.txt @@ -21,4 +21,19 @@ 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] \ No newline at end of file +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]