Update array_api_compat/common/_helpers.py

crusaderky · jakevdp · web-flow · commit 05680430ad57 · 2024-11-26T14:55:07.000Z
Co-authored-by: Jake Vanderplas &lt;jakevdp@gmail.com&gt;
diff --git a/array_api_compat/common/_helpers.py b/array_api_compat/common/_helpers.py
@@ -856,7 +856,7 @@ class at:
 
     If x is a JAX array, they are the same as::
 
-      x = x.at[1].add(x)
+      x = x.at[1].add(2)
 
     If x is a read-only numpy array, they are the same as::
 
