@@ -5,7 +5,7 @@ import numpy as np
55import optype .numpy as onp
66import optype .numpy .compat as npc
77
8- from scipy .sparse . _base import _SpArray2D , _SpMatrix
8+ from scipy .sparse import sparray , spmatrix
99from 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