Fix undefined variable errors

jax-ml · Sep 27, 2024 · f69d18e · f69d18e
1 parent 88b3350
commit f69d18e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/jax/_src/numpy/ufuncs.py b/jax/_src/numpy/ufuncs.py
@@ -1015,6 +1015,7 @@ def arctanh(x: ArrayLike, /) -> Array:
 
     For complex-valued input:
 
+    >>> x1 = jnp.array([-2+0j, 3+0j, 4-1j])
     >>> with jnp.printoptions(precision=3, suppress=True):
     ...   jnp.arctanh(x1)
     Array([-0.549+1.571j,  0.347+1.571j,  0.239-1.509j], dtype=complex64)