Merge pull request #25952 from johannahaffner:patch-1

PiperOrigin-RevId: 716678666
jax-ml · Jan 17, 2025 · a4a657b · a4a657b
2 parents a527aba + df6140e
commit a4a657b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax/_src/numpy/lax_numpy.py b/jax/_src/numpy/lax_numpy.py
@@ -12894,7 +12894,7 @@ def cov(m: ArrayLike, y: ArrayLike | None = None, rowvar: bool = True,
       observation.
 
   Returns:
-    A covariance matrix of shape ``(M, M)``.
+    A covariance matrix of shape ``(M, M)``, or a scalar with shape ``()`` if ``M = 1``.
 
   See also:
     - :func:`jax.numpy.corrcoef`: compute the correlation coefficient, a normalized