Skip to content

Commit cd50df2

Browse files
committed
🎨 cleaned up type aliases
1 parent 32a853d commit cd50df2

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

‎scipy-stubs/sparse/linalg/_eigen/arpack/arpack.pyi‎

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ import numpy as np
55
import optype.numpy as onp
66
import optype.numpy.compat as npc
77

8-
from scipy.sparse._base import _SpArray2D, _SpMatrix
8+
from scipy.sparse import sparray, spmatrix
99
from scipy.sparse.linalg import LinearOperator
1010

1111
__all__ = ["ArpackError", "ArpackNoConvergence", "eigs", "eigsh"]
@@ -15,13 +15,12 @@ _SCT = TypeVar("_SCT", bound=_Numeric, default=_Numeric)
1515

1616
_Numeric: TypeAlias = npc.number | np.bool_
1717
_ToFloat: TypeAlias = npc.floating | npc.integer | np.bool_
18-
_ToJustComplex: TypeAlias = npc.complexfloating
1918

20-
_Sparse2D: TypeAlias = _SpMatrix[_SCT] | _SpArray2D[_SCT]
19+
_MatrixOperator: TypeAlias = spmatrix[_SCT] | sparray[_SCT, tuple[int, int]] | LinearOperator[_SCT]
2120

22-
_ToRealMatrix: TypeAlias = onp.ToFloat2D | _Sparse2D[_ToFloat] | LinearOperator[_ToFloat]
23-
_ToJustComplexMatrix: TypeAlias = onp.ToJustComplex2D | _Sparse2D[_ToJustComplex] | LinearOperator[_ToJustComplex]
24-
_ToComplexMatrix: TypeAlias = onp.ToComplex2D | _Sparse2D | LinearOperator
21+
_ToRealMatrix: TypeAlias = onp.ToFloat2D | _MatrixOperator[_ToFloat]
22+
_ToJustComplexMatrix: TypeAlias = onp.ToJustComplex2D | _MatrixOperator[npc.complexfloating]
23+
_ToComplexMatrix: TypeAlias = onp.ToComplex2D | _MatrixOperator
2524

2625
_Which_eigs: TypeAlias = Literal["LM", "SM", "LR", "SR", "LI", "SI"]
2726
_Which_eigsh: TypeAlias = Literal["LM", "SM", "LA", "SA", "BE"]

0 commit comments

Comments
 (0)