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

PiperOrigin-RevId: 660941835

.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: '33f2d2ea2f3dd2b3ceeeb4519d55e08096184149'  # Latest commit as of 2024-05-28
+        ref: 'db95e67b29235249e5776ca2b6bb4e77117e0690'  # Latest commit as of 2024-08-08
         submodules: 'true'
         path: 'array-api-tests'
     - name: Set up Python ${{ matrix.python-version }}

pyproject.toml

@@ -60,7 +60,8 @@ filterwarnings = [
     # TODO(jakevdp): remove when array_api_tests stabilize
     "default:.*not machine-readable.*:UserWarning",
     "default:Special cases found for .* but none were parsed.*:UserWarning",
-    "default:.*is not JSON-serializable. Using the repr instead.",
+    "default:.*is not JSON-serializable. Using the repr instead.*:UserWarning",
+    "default:The .* method is good for exploring strategies.*",
 
     # These are transitive warnings coming from TensorFlow dependencies.
     # TODO(slebedev): Remove once we bump the minimum TensorFlow version.

tests/array_api_skips.txt

@@ -4,36 +4,11 @@
 array_api_tests/test_data_type_functions.py::test_finfo[float32]
 
 # Test suite attempts in-place mutation:
-array_api_tests/test_special_cases.py::test_iop
-array_api_tests/test_special_cases.py::test_nan_propagation
 array_api_tests/test_array_object.py::test_setitem
+array_api_tests/test_array_object.py::test_setitem_masking
 
-# Raises NonInteractiveExampleWarning
-array_api_tests/test_special_cases.py::test_binary
-array_api_tests/test_special_cases.py::test_unary
-
-# 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
-
-# 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]
+# Returns wrong zero sign
+array_api_tests/test_special_cases.py::test_unary[sign((x_i is -0 or x_i == +0)) -> 0]
 
 # Returns int32 when int64 is expected
 array_api_tests/test_searching_functions.py::test_searchsorted
-
-# Various info functions not yet defined
-# Pending bugfix, see https://github.com/data-apis/array-api-tests/pull/262
-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]