From 3bbd141d3d5205f7554630b485c9e9a58affae2c Mon Sep 17 00:00:00 2001 From: tilakrayal <81610181+tilakrayal@users.noreply.github.com> Date: Thu, 27 Jun 2024 13:22:39 +0530 Subject: [PATCH] Fixing the naming conventions in linalg.py --- jax/_src/scipy/linalg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jax/_src/scipy/linalg.py b/jax/_src/scipy/linalg.py index 72f07a3441eb..5458d71dedf4 100644 --- a/jax/_src/scipy/linalg.py +++ b/jax/_src/scipy/linalg.py @@ -415,7 +415,7 @@ def eigh(a: ArrayLike, b: ArrayLike | None = None, lower: bool = True, type: int = 1, check_finite: bool = True) -> Array | tuple[Array, Array]: """Compute eigenvalues and eigenvectors for a Hermitian matrix - JAX implementation of :func:`jax.scipy.linalg.eigh`. + JAX implementation of :func:`scipy.linalg.eigh`. Args: a: Hermitian input array of shape ``(..., N, N)``