From a4c249653ec9d063a67a6cde8123dca2defb8f8b Mon Sep 17 00:00:00 2001 From: Bas van Beek Date: Wed, 17 May 2023 21:16:35 +0200 Subject: TYP: Update type annotations for the new linalg named tuples --- numpy/linalg/linalg.py | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) (limited to 'numpy/linalg/linalg.py') diff --git a/numpy/linalg/linalg.py b/numpy/linalg/linalg.py index b939c9c95..0f06c8520 100644 --- a/numpy/linalg/linalg.py +++ b/numpy/linalg/linalg.py @@ -17,7 +17,7 @@ __all__ = ['matrix_power', 'solve', 'tensorsolve', 'tensorinv', 'inv', import functools import operator import warnings -from typing import NamedTuple +from typing import NamedTuple, Any from .._utils import set_module from numpy.core import ( @@ -38,24 +38,24 @@ from numpy._typing import NDArray class EigResult(NamedTuple): eigenvalues: NDArray[Any] - eigenvectors: NDArray + eigenvectors: NDArray[Any] class EighResult(NamedTuple): - eigenvalues: NDArray - eigenvectors: NDArray + eigenvalues: NDArray[Any] + eigenvectors: NDArray[Any] class QRResult(NamedTuple): - Q: NDArray - R: NDArray + Q: NDArray[Any] + R: NDArray[Any] class SlogdetResult(NamedTuple): - sign: NDArray - logabsdet: NDArray + sign: NDArray[Any] + logabsdet: NDArray[Any] class SVDResult(NamedTuple): - U: NDArray - S: NDArray - Vh: NDArray + U: NDArray[Any] + S: NDArray[Any] + Vh: NDArray[Any] array_function_dispatch = functools.partial( overrides.array_function_dispatch, module='numpy.linalg') -- cgit v1.2.1