From 3a4d0b35529672c5c448d70a42144ef4e0b8b155 Mon Sep 17 00:00:00 2001
From: jiayaobo <januzaj2011@163.com>
Date: Wed, 29 Mar 2023 11:39:49 +0800
Subject: [PATCH] remove scale in wald docstring

---
 jax/_src/random.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/jax/_src/random.py b/jax/_src/random.py
index d8238c4e555c..1f2a46a9dc5a 100644
--- a/jax/_src/random.py
+++ b/jax/_src/random.py
@@ -2069,7 +2069,7 @@ def wald(key: KeyArray,
 
   Returns:
     A random array with the specified dtype and with shape given by ``shape`` if
-    ``shape`` is not None, or else by ``mean.shape`` and ``scale.shape``.
+    ``shape`` is not None, or else by ``mean.shape``.
   """
   key, _ = _check_prng_key(key)
   if not dtypes.issubdtype(dtype, np.floating):