From 6fb357d651ca39448ae2db94f8abc5d16a7a702b Mon Sep 17 00:00:00 2001 From: Jake VanderPlas Date: Wed, 29 May 2024 08:48:49 -0700 Subject: [PATCH] [array api] update test suite to most recent commit --- .github/workflows/jax-array-api.yml | 2 +- jax/experimental/array_api/skips.txt | 21 ++++++++++++++++----- 2 files changed, 17 insertions(+), 6 deletions(-) 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..6da8afb6591e 100644 --- a/jax/experimental/array_api/skips.txt +++ b/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] \ 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]