From eb6b721580efd9461528abb3d2c4966365c5b367 Mon Sep 17 00:00:00 2001 From: Guido Imperiale Date: Sat, 25 Jan 2025 13:40:28 +0000 Subject: [PATCH] Update src/array_api_extra/_lib/_at.py Co-authored-by: Lucas Colley --- src/array_api_extra/_lib/_at.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/array_api_extra/_lib/_at.py b/src/array_api_extra/_lib/_at.py index e0dab54..e5cb387 100644 --- a/src/array_api_extra/_lib/_at.py +++ b/src/array_api_extra/_lib/_at.py @@ -148,7 +148,7 @@ class at: # pylint: disable=invalid-name # numpydoc ignore=PR02 Boolean masks are supported on Dask and jitted JAX arrays exclusively when `idx` has the same shape as `x` and `y` is 0-dimensional. - Note that this is support is not available in JAX's native + Note that this support is not available in JAX's native ``x.at[mask].set(y)``. This pattern::