@@ -25,7 +25,6 @@ from ._typing import (
25
25
Float ,
26
26
Index1D ,
27
27
Int ,
28
- Matrix ,
29
28
Scalar ,
30
29
Shape ,
31
30
ShapeCOO ,
@@ -161,12 +160,12 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
161
160
@overload # dense array-like int
162
161
def __init__ (self : _spbase [np .int_ ], / , arg1 : _ToSparseFromPy [opt .JustInt ], * , maxprint : int | None = 50 ) -> None : ...
163
162
@overload # dense array-like float
164
- def __init__ (self : _spbase [np .float64 ], / , arg1 : _ToSparseFromPy [opt .Just [ float ] ], * , maxprint : int | None = 50 ) -> None : ...
163
+ def __init__ (self : _spbase [np .float64 ], / , arg1 : _ToSparseFromPy [opt .JustFloat ], * , maxprint : int | None = 50 ) -> None : ...
165
164
@overload # dense array-like cfloat
166
165
def __init__ (
167
166
self : _spbase [np .complex128 ],
168
167
/ ,
169
- arg1 : _ToSparseFromPy [opt .Just [ complex ] ],
168
+ arg1 : _ToSparseFromPy [opt .JustComplex ],
170
169
* ,
171
170
maxprint : int | None = 50 ,
172
171
) -> None : ...
@@ -272,11 +271,11 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
272
271
@overload # sparse[-Complex], sparse[+Complex64]
273
272
def __add__ (self : _SpFromComplexT , other : _spbase [_ToComplex64 ], / ) -> _SpFromComplexT : ...
274
273
@overload # spmatrix[-Int], array[+Int8]
275
- def __add__ (self : spmatrix [_FromInt ], other : onp .ArrayND [_ToInt8 ], / ) -> Matrix [_SCT_co ]: ...
274
+ def __add__ (self : spmatrix [_FromInt ], other : onp .ArrayND [_ToInt8 ], / ) -> onp . Matrix [_SCT_co ]: ...
276
275
@overload # spmatrix[-Float], array[+Float32]
277
- def __add__ (self : spmatrix [_FromFloat ], other : onp .ArrayND [_ToFloat32 ], / ) -> Matrix [_SCT_co ]: ...
276
+ def __add__ (self : spmatrix [_FromFloat ], other : onp .ArrayND [_ToFloat32 ], / ) -> onp . Matrix [_SCT_co ]: ...
278
277
@overload # spmatrix[-Complex], array[+Complex64]
279
- def __add__ (self : spmatrix [_FromComplex ], other : onp .ArrayND [_ToComplex64 ], / ) -> Matrix [_SCT_co ]: ...
278
+ def __add__ (self : spmatrix [_FromComplex ], other : onp .ArrayND [_ToComplex64 ], / ) -> onp . Matrix [_SCT_co ]: ...
280
279
@overload # sparse[-Int], array[+Int8]
281
280
def __add__ (self : _spbase [_FromInt ], other : onp .ArrayND [_ToInt8 ], / ) -> onp .ArrayND [_SCT_co , _ShapeT_co ]: ...
282
281
@overload # sparse[-Float], array[+Float32]
@@ -297,11 +296,11 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
297
296
@overload # sparse[-Complex], sparse[+Complex64]
298
297
def __sub__ (self : _SpFromComplexT , other : _spbase [_ToComplex64 ], / ) -> _SpFromComplexT : ...
299
298
@overload # spmatrix[-Int], array[+Int8]
300
- def __sub__ (self : spmatrix [_FromInt ], other : onp .ArrayND [_ToInt8 ], / ) -> Matrix [_SCT_co ]: ...
299
+ def __sub__ (self : spmatrix [_FromInt ], other : onp .ArrayND [_ToInt8 ], / ) -> onp . Matrix [_SCT_co ]: ...
301
300
@overload # spmatrix[-Float], array[+Float32]
302
- def __sub__ (self : spmatrix [_FromFloat ], other : onp .ArrayND [_ToFloat32 ], / ) -> Matrix [_SCT_co ]: ...
301
+ def __sub__ (self : spmatrix [_FromFloat ], other : onp .ArrayND [_ToFloat32 ], / ) -> onp . Matrix [_SCT_co ]: ...
303
302
@overload # spmatrix[-Complex], array[+Complex64]
304
- def __sub__ (self : spmatrix [_FromComplex ], other : onp .ArrayND [_ToComplex64 ], / ) -> Matrix [_SCT_co ]: ...
303
+ def __sub__ (self : spmatrix [_FromComplex ], other : onp .ArrayND [_ToComplex64 ], / ) -> onp . Matrix [_SCT_co ]: ...
305
304
@overload # sparse[-Int], array[+Int8]
306
305
def __sub__ (self : _spbase [_FromInt ], other : onp .ArrayND [_ToInt8 ], / ) -> onp .ArrayND [_SCT_co , _ShapeT_co ]: ...
307
306
@overload # sparse[-Float], array[+Float32]
@@ -360,13 +359,13 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
360
359
@overload # spmatrix[+Bool], /, other: array-like ~Int
361
360
def __mul__ (self : spmatrix [_ToBool ], / , other : _DualArrayLike2d [opt .JustInt , Int ]) -> onp .Array2D [Int ]: ...
362
361
@overload # spmatrix[+Int], /, other: scalar- or matrix-like ~Float
363
- def __mul__ (self : spmatrix [_ToInt ], / , other : _DualMatrixLike [opt .Just [ float ] , Float ]) -> spmatrix [Float ]: ...
362
+ def __mul__ (self : spmatrix [_ToInt ], / , other : _DualMatrixLike [opt .JustFloat , Float ]) -> spmatrix [Float ]: ...
364
363
@overload # spmatrix[+Int], /, other: array-like ~Float
365
- def __mul__ (self : spmatrix [_ToInt ], / , other : _DualArrayLike2d [opt .Just [ float ] , Float ]) -> onp .Array2D [Float ]: ...
364
+ def __mul__ (self : spmatrix [_ToInt ], / , other : _DualArrayLike2d [opt .JustFloat , Float ]) -> onp .Array2D [Float ]: ...
366
365
@overload # spmatrix[+Float], /, other: scalar- or matrix-like ~Complex
367
- def __mul__ (self : spmatrix [_ToFloat ], / , other : _DualMatrixLike [opt .Just [ complex ] , Complex ]) -> spmatrix [Complex ]: ...
366
+ def __mul__ (self : spmatrix [_ToFloat ], / , other : _DualMatrixLike [opt .JustComplex , Complex ]) -> spmatrix [Complex ]: ...
368
367
@overload # spmatrix[+Float], /, other: array-like ~Complex
369
- def __mul__ (self : spmatrix [_ToFloat ], / , other : _DualArrayLike2d [opt .Just [ complex ] , Complex ]) -> onp .Array2D [Complex ]: ...
368
+ def __mul__ (self : spmatrix [_ToFloat ], / , other : _DualArrayLike2d [opt .JustComplex , Complex ]) -> onp .Array2D [Complex ]: ...
370
369
@overload # Self[+Bool], /, other: -Int
371
370
def __mul__ (self : _spbase [_ToBool ], / , other : _FromIntT ) -> _spbase [_FromIntT , _ShapeT_co ]: ...
372
371
@overload # Self[+Int], /, other: -Float
@@ -424,11 +423,11 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
424
423
@overload # spmatrix[+Int], other: scalar- or matrix-like ~Float
425
424
def __matmul__ (self : spmatrix [_ToInt ], other : _spbase [Float ], / ) -> _SpMatrixOut [Float ]: ...
426
425
@overload # spmatrix[+Int], other: array-like ~Float
427
- def __matmul__ (self : spmatrix [_ToInt ], other : _DualArrayLike2d [opt .Just [ float ] , Float ], / ) -> onp .Array2D [Float ]: ...
426
+ def __matmul__ (self : spmatrix [_ToInt ], other : _DualArrayLike2d [opt .JustFloat , Float ], / ) -> onp .Array2D [Float ]: ...
428
427
@overload # spmatrix[+Float], other: scalar- or matrix-like ~Complex
429
428
def __matmul__ (self : spmatrix [_ToFloat ], other : _spbase [Complex ], / ) -> _SpMatrixOut [Complex ]: ...
430
429
@overload # spmatrix[+Float], other: array-like ~Complex
431
- def __matmul__ (self : spmatrix [_ToFloat ], other : _DualArrayLike2d [opt .Just [ complex ] , Complex ], / ) -> onp .Array2D [Complex ]: ...
430
+ def __matmul__ (self : spmatrix [_ToFloat ], other : _DualArrayLike2d [opt .JustComplex , Complex ], / ) -> onp .Array2D [Complex ]: ...
432
431
@overload # catch-all
433
432
def __matmul__ (self , other : _DualArrayLike [complex , Scalar ] | _spbase , / ) -> _spbase [Any , Any ] | onp .Array [Any , Any ]: ...
434
433
__rmatmul__ = __matmul__
@@ -449,11 +448,11 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
449
448
@overload
450
449
def __truediv__ (self : spmatrix , rhs : onp .ArrayND [Scalar ], / ) -> coo_matrix [np .int_ | _FromFloat ]: ...
451
450
@overload
452
- def __truediv__ (self : spmatrix , rhs : _spbase [_FromComplex ], / ) -> Matrix [_FromComplex ]: ...
451
+ def __truediv__ (self : spmatrix , rhs : _spbase [_FromComplex ], / ) -> onp . Matrix [_FromComplex ]: ...
453
452
@overload
454
- def __truediv__ (self : spmatrix , rhs : _spbase [_FromFloat ], / ) -> Matrix [_FromFloat ]: ...
453
+ def __truediv__ (self : spmatrix , rhs : _spbase [_FromFloat ], / ) -> onp . Matrix [_FromFloat ]: ...
455
454
@overload
456
- def __truediv__ (self : spmatrix , rhs : _spbase , / ) -> Matrix [np .int_ | _FromFloat ]: ...
455
+ def __truediv__ (self : spmatrix , rhs : _spbase , / ) -> onp . Matrix [np .int_ | _FromFloat ]: ...
457
456
@overload
458
457
def __truediv__ (self , rhs : _spbase [_FromComplex ], / ) -> onp .ArrayND [_FromComplex , _ShapeT_co ]: ...
459
458
@overload
@@ -477,9 +476,9 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
477
476
@overload # spmatrix[+Bool], /, other: scalar-like ~Int
478
477
def dot (self : spmatrix [_ToBool ], / , other : opt .JustInt ) -> spmatrix [Int ]: ...
479
478
@overload # spmatrix[+Int], /, other: scalar-like ~Float
480
- def dot (self : spmatrix [_ToInt ], / , other : opt .Just [ float ] ) -> spmatrix [Float ]: ...
479
+ def dot (self : spmatrix [_ToInt ], / , other : opt .JustFloat ) -> spmatrix [Float ]: ...
481
480
@overload # spmatrix[+Float], /, other: scalar-like ~Complex
482
- def dot (self : spmatrix [_ToFloat ], / , other : opt .Just [ complex ] ) -> spmatrix [Complex ]: ...
481
+ def dot (self : spmatrix [_ToFloat ], / , other : opt .JustComplex ) -> spmatrix [Complex ]: ...
483
482
@overload # sparray[-Bool], /, other: sparse +Bool
484
483
def dot (self : _SpArray , / , other : _spbase [_ToBool | _SCT_co ]) -> _SpArrayOut [_SCT_co ]: ...
485
484
@overload # sparray[-Bool], /, other: array-like +Bool
@@ -521,11 +520,11 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
521
520
@overload # spmatrix[+Int], /, other: scalar- or matrix-like ~Float
522
521
def dot (self : spmatrix [_ToInt ], / , other : _spbase [Float ]) -> _SpMatrixOut [Float ]: ...
523
522
@overload # spmatrix[+Int], /, other: array-like ~Float
524
- def dot (self : spmatrix [_ToInt ], / , other : _DualArrayLike2d [opt .Just [ float ] , Float ]) -> onp .Array2D [Float ]: ...
523
+ def dot (self : spmatrix [_ToInt ], / , other : _DualArrayLike2d [opt .JustFloat , Float ]) -> onp .Array2D [Float ]: ...
525
524
@overload # spmatrix[+Float], /, other: scalar- or matrix-like ~Complex
526
525
def dot (self : spmatrix [_ToFloat ], / , other : _spbase [Complex ]) -> _SpMatrixOut [Complex ]: ...
527
526
@overload # spmatrix[+Float], /, other: array-like ~Complex
528
- def dot (self : spmatrix [_ToFloat ], / , other : _DualArrayLike2d [opt .Just [ complex ] , Complex ]) -> onp .Array2D [Complex ]: ...
527
+ def dot (self : spmatrix [_ToFloat ], / , other : _DualArrayLike2d [opt .JustComplex , Complex ]) -> onp .Array2D [Complex ]: ...
529
528
@overload # catch-all
530
529
def dot (self , / , other : _DualArrayLike [complex , Scalar ] | _spbase ) -> _spbase [Any , Any ] | onp .Array [Any , Any ]: ...
531
530
@@ -551,7 +550,7 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
551
550
@overload # sparray, dense
552
551
def maximum (self : sparray , / , other : onp .ArrayND [_SCT ]) -> onp .ArrayND [_SCT_co | _SCT , _ShapeT_co ]: ...
553
552
@overload # spmatrix, dense
554
- def maximum (self : spmatrix , / , other : onp .ArrayND [_SCT ]) -> Matrix [_SCT_co | _SCT ]: ...
553
+ def maximum (self : spmatrix , / , other : onp .ArrayND [_SCT ]) -> onp . Matrix [_SCT_co | _SCT ]: ...
555
554
@overload # sparray, unknown scalar
556
555
def maximum (self : sparray , / , other : onp .ToComplex ) -> _SpArrayOut [Any ]: ...
557
556
@overload # spmatrix, unknown scalar
@@ -579,7 +578,7 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
579
578
@overload # sparray, dense
580
579
def minimum (self : sparray , / , other : onp .ArrayND [_SCT ]) -> onp .ArrayND [_SCT_co | _SCT , _ShapeT_co ]: ...
581
580
@overload # spmatrix, dense
582
- def minimum (self : spmatrix , / , other : onp .ArrayND [_SCT ]) -> Matrix [_SCT_co | _SCT ]: ...
581
+ def minimum (self : spmatrix , / , other : onp .ArrayND [_SCT ]) -> onp . Matrix [_SCT_co | _SCT ]: ...
583
582
@overload # sparray, unknown scalar
584
583
def minimum (self : sparray , / , other : onp .ToComplex ) -> _SpArrayOut [Any ]: ...
585
584
@overload # spmatrix, unknown scalar
@@ -622,7 +621,7 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
622
621
@overload # sparray[-Bool, 2d], axis: index
623
622
def sum (self : _SpArray2D [np .bool_ ], / , axis : op .CanIndex , dtype : None = None , out : None = None ) -> onp .Array1D [np .int_ ]: ...
624
623
@overload # spmatrix[-Bool], axis: index
625
- def sum (self : spmatrix [np .bool_ ], / , axis : op .CanIndex , dtype : None = None , out : None = None ) -> Matrix [np .int_ ]: ...
624
+ def sum (self : spmatrix [np .bool_ ], / , axis : op .CanIndex , dtype : None = None , out : None = None ) -> onp . Matrix [np .int_ ]: ...
626
625
@overload # Self[-Int]
627
626
def sum (self : _spbase [_FromIntT ], / , axis : None = None , dtype : None = None , out : None = None ) -> _FromIntT : ...
628
627
@overload # sparray[-Int, 1d], axis: index
@@ -642,9 +641,9 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
642
641
out : None = None ,
643
642
) -> onp .Array1D [_FromIntT ]: ...
644
643
@overload # spmatrix[-Int], axis: index
645
- def sum (self : spmatrix [_FromIntT ], / , axis : op .CanIndex , dtype : None = None , out : None = None ) -> Matrix [_FromIntT ]: ...
644
+ def sum (self : spmatrix [_FromIntT ], / , axis : op .CanIndex , dtype : None = None , out : None = None ) -> onp . Matrix [_FromIntT ]: ...
646
645
@overload # spmatrix, axis: index, dtype: <unknown>
647
- def sum (self : spmatrix , / , axis : op .CanIndex , dtype : npt .DTypeLike , out : None = None ) -> Matrix [Any ]: ...
646
+ def sum (self : spmatrix , / , axis : op .CanIndex , dtype : npt .DTypeLike , out : None = None ) -> onp . Matrix [Any ]: ...
648
647
@overload # dtype: <unknown> (keyword)
649
648
def sum (self , / , axis : op .CanIndex | None = None , * , dtype : npt .DTypeLike , out : None = None ) -> Any : ... # noqa: ANN401
650
649
@@ -680,7 +679,7 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
680
679
axis : op .CanIndex ,
681
680
dtype : None = None ,
682
681
out : None = None ,
683
- ) -> Matrix [np .float64 ]: ...
682
+ ) -> onp . Matrix [np .float64 ]: ...
684
683
@overload # sparray[-Float, 1d]
685
684
def mean (
686
685
self : _SpArray1D [_FromFloatT ],
@@ -698,9 +697,11 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
698
697
out : None = None ,
699
698
) -> onp .Array1D [_FromFloatT ]: ...
700
699
@overload # spmatrix[-Float], axis: index
701
- def mean (self : spmatrix [_FromFloatT ], / , axis : op .CanIndex , dtype : None = None , out : None = None ) -> Matrix [_FromFloatT ]: ...
700
+ def mean (
701
+ self : spmatrix [_FromFloatT ], / , axis : op .CanIndex , dtype : None = None , out : None = None
702
+ ) -> onp .Matrix [_FromFloatT ]: ...
702
703
@overload # spmatrix, axis: index, dtype: <unknown>
703
- def mean (self : spmatrix , / , axis : op .CanIndex , dtype : npt .DTypeLike , out : None = None ) -> Matrix [Any ]: ...
704
+ def mean (self : spmatrix , / , axis : op .CanIndex , dtype : npt .DTypeLike , out : None = None ) -> onp . Matrix [Any ]: ...
704
705
@overload # dtype: <unknown> (keyword)
705
706
def mean (self , / , axis : op .CanIndex | None = None , * , dtype : npt .DTypeLike , out : None = None ) -> Any : ... # noqa: ANN401
706
707
@@ -925,11 +926,11 @@ class _spbase(Generic[_SCT_co, _ShapeT_co]):
925
926
926
927
#
927
928
@overload # self: spmatrix, out: None
928
- def todense (self : spmatrix , / , order : OrderCF | None = None , out : None = None ) -> Matrix [_SCT_co ]: ...
929
+ def todense (self : spmatrix , / , order : OrderCF | None = None , out : None = None ) -> onp . Matrix [_SCT_co ]: ...
929
930
@overload # self: spmatrix, out: array (positional)
930
- def todense (self : spmatrix , / , order : OrderCF | None , out : onp .ArrayND [_SCT ]) -> Matrix [_SCT ]: ...
931
+ def todense (self : spmatrix , / , order : OrderCF | None , out : onp .ArrayND [_SCT ]) -> onp . Matrix [_SCT ]: ...
931
932
@overload # self: spmatrix, out: array (keyword)
932
- def todense (self : spmatrix , / , order : OrderCF | None = None , * , out : onp .ArrayND [_SCT ]) -> Matrix [_SCT ]: ...
933
+ def todense (self : spmatrix , / , order : OrderCF | None = None , * , out : onp .ArrayND [_SCT ]) -> onp . Matrix [_SCT ]: ...
933
934
@overload # out: None
934
935
def todense (self , / , order : OrderCF | None = None , out : None = None ) -> onp .Array [_ShapeT_co , _SCT_co ]: ...
935
936
@overload # out: array (positional)
0 commit comments