remove scale in wald docstring

jax-ml · Mar 29, 2023 · 3a4d0b3 · 3a4d0b3
1 parent c2d6fcc
commit 3a4d0b3
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):