Merge pull request #15283 from JiaYaobo:fix_wald_doc

PiperOrigin-RevId: 520364879
jax-ml · Mar 29, 2023 · cfa330b · cfa330b
2 parents 2d94f76 + 3a4d0b3
commit cfa330b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):