Skip to content

Commit 0ec1442

Browse files
JulVandenBroeckJulVandenBroeck
committed
Added DOK stubs + merged dtype=None and dtype=np.float64 cases
1 parent 2fe7a0d commit 0ec1442

File tree

7 files changed

+87
-120
lines changed

7 files changed

+87
-120
lines changed

‎scipy-stubs/sparse/_bsr.pyi

Lines changed: 3 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -98,18 +98,18 @@ class bsr_array(_bsr_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G
9898
*,
9999
maxprint: int | None = None,
100100
) -> None: ...
101-
@overload # 2-d shape-like, dtype: None
101+
@overload # 2-d shape-like, dtype: type[float] | None
102102
def __init__(
103103
self: bsr_array[np.float64],
104104
/,
105105
arg1: ToShape2D,
106106
shape: ToShape2D | None = None,
107-
dtype: None = None,
107+
dtype: onp.AnyFloat64DType | None = None,
108108
copy: bool = False,
109109
*,
110110
maxprint: int | None = None,
111111
) -> None: ...
112-
@overload # 2-d shape-like, dtype: type[bool]
112+
@overload # 2-d shape-like, dtype: type[bool] | None
113113
def __init__(
114114
self: bsr_array[np.bool_],
115115
/,
@@ -131,17 +131,6 @@ class bsr_array(_bsr_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G
131131
*,
132132
maxprint: int | None = None,
133133
) -> None: ...
134-
@overload # 2-d shape-like, dtype: type[float]
135-
def __init__(
136-
self: bsr_array[np.float64],
137-
/,
138-
arg1: ToShape2D,
139-
shape: ToShape2D | None = None,
140-
dtype: onp.AnyFloat64DType | None = None,
141-
copy: bool = False,
142-
*,
143-
maxprint: int | None = None,
144-
) -> None: ...
145134
@overload # 2-d shape-like, dtype: type[complex]
146135
def __init__(
147136
self: bsr_array[np.complex128],

‎scipy-stubs/sparse/_coo.pyi

Lines changed: 6 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -149,35 +149,35 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT
149149
*,
150150
maxprint: int | None = None,
151151
) -> None: ...
152-
@overload # 1-d shape-like, dtype: None
152+
@overload # 1-d shape-like, dtype: type[float] | None
153153
def __init__(
154154
self: coo_array[np.float64, tuple[int]],
155155
/,
156156
arg1: ToShape1D,
157157
shape: ToShape1D | None = None,
158-
dtype: None = None,
158+
dtype: onp.AnyFloat64DType | None = None,
159159
copy: bool = False,
160160
*,
161161
maxprint: int | None = None,
162162
) -> None: ...
163-
@overload # 2-d shape-like, dtype: None
163+
@overload # 2-d shape-like, dtype: type[float] | None
164164
def __init__(
165165
self: coo_array[np.float64, tuple[int, int]],
166166
/,
167167
arg1: ToShape2D,
168168
shape: ToShape2D | None = None,
169-
dtype: None = None,
169+
dtype: onp.AnyFloat64DType | None = None,
170170
copy: bool = False,
171171
*,
172172
maxprint: int | None = None,
173173
) -> None: ...
174-
@overload # >2-d shape-like, dtype: None
174+
@overload # >2-d shape-like, dtype: type[float]
175175
def __init__(
176176
self: coo_array[np.float64, onp.AtLeast3D],
177177
/,
178178
arg1: ToShapeMin3D,
179179
shape: ToShapeMin3D | None = None,
180-
dtype: None = None,
180+
dtype: onp.AnyFloat64DType | None = None,
181181
copy: bool = False,
182182
*,
183183
maxprint: int | None = None,
@@ -248,39 +248,6 @@ class coo_array(_coo_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT
248248
*,
249249
maxprint: int | None = None,
250250
) -> None: ...
251-
@overload # 1-d shape-like, dtype: type[float]
252-
def __init__(
253-
self: coo_array[np.float64, tuple[int]],
254-
/,
255-
arg1: ToShape1D,
256-
shape: ToShape1D | None = None,
257-
dtype: onp.AnyFloat64DType | None = None,
258-
copy: bool = False,
259-
*,
260-
maxprint: int | None = None,
261-
) -> None: ...
262-
@overload # 2-d shape-like, dtype: type[float]
263-
def __init__(
264-
self: coo_array[np.float64, tuple[int, int]],
265-
/,
266-
arg1: ToShape2D,
267-
shape: ToShape2D | None = None,
268-
dtype: onp.AnyFloat64DType | None = None,
269-
copy: bool = False,
270-
*,
271-
maxprint: int | None = None,
272-
) -> None: ...
273-
@overload # >2-d shape-like, dtype: type[float]
274-
def __init__(
275-
self: coo_array[np.float64, onp.AtLeast3D],
276-
/,
277-
arg1: ToShapeMin3D,
278-
shape: ToShapeMin3D | None = None,
279-
dtype: onp.AnyFloat64DType | None = None,
280-
copy: bool = False,
281-
*,
282-
maxprint: int | None = None,
283-
) -> None: ...
284251
@overload # 1-d shape-like, dtype: type[complex]
285252
def __init__(
286253
self: coo_array[np.complex128, tuple[int]],

‎scipy-stubs/sparse/_csc.pyi

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -77,13 +77,13 @@ class csc_array(_csc_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G
7777
*,
7878
maxprint: int | None = None,
7979
) -> None: ...
80-
@overload # 2-d shape-like, dtype: None
80+
@overload # 2-d shape-like, dtype: type[float] | None
8181
def __init__(
8282
self: csc_array[np.float64],
8383
/,
8484
arg1: ToShape2D,
8585
shape: ToShape2D | None = None,
86-
dtype: None = None,
86+
dtype: onp.AnyFloat64DType | None = None,
8787
copy: bool = False,
8888
*,
8989
maxprint: int | None = None,
@@ -110,17 +110,6 @@ class csc_array(_csc_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G
110110
*,
111111
maxprint: int | None = None,
112112
) -> None: ...
113-
@overload # 2-d shape-like, dtype: type[float]
114-
def __init__(
115-
self: csc_array[np.float64],
116-
/,
117-
arg1: ToShape2D,
118-
shape: ToShape2D | None = None,
119-
dtype: onp.AnyFloat64DType | None = None,
120-
copy: bool = False,
121-
*,
122-
maxprint: int | None = None,
123-
) -> None: ...
124113
@overload # 2-d shape-like, dtype: type[complex]
125114
def __init__(
126115
self: csc_array[np.complex128],

‎scipy-stubs/sparse/_csr.pyi

Lines changed: 4 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -120,24 +120,24 @@ class csr_array(_csr_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT
120120
*,
121121
maxprint: int | None = None,
122122
) -> None: ...
123-
@overload # 1-d shape-like, dtype: None
123+
@overload # 1-d shape-like, dtype: type[float] | None
124124
def __init__(
125125
self: csr_array[np.float64, tuple[int]],
126126
/,
127127
arg1: ToShape1D,
128128
shape: ToShape1D | None = None,
129-
dtype: None = None,
129+
dtype: onp.AnyFloat64DType | None = None,
130130
copy: bool = False,
131131
*,
132132
maxprint: int | None = None,
133133
) -> None: ...
134-
@overload # 2-d shape-like, dtype: None
134+
@overload # 2-d shape-like, dtype: type[float] | None
135135
def __init__(
136136
self: csr_array[np.float64, tuple[int, int]],
137137
/,
138138
arg1: ToShape2D,
139139
shape: ToShape2D | None = None,
140-
dtype: None = None,
140+
dtype: onp.AnyFloat64DType | None = None,
141141
copy: bool = False,
142142
*,
143143
maxprint: int | None = None,
@@ -186,28 +186,6 @@ class csr_array(_csr_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT
186186
*,
187187
maxprint: int | None = None,
188188
) -> None: ...
189-
@overload # 1-d shape-like, dtype: type[float]
190-
def __init__(
191-
self: csr_array[np.float64, tuple[int]],
192-
/,
193-
arg1: ToShape1D,
194-
shape: ToShape1D | None = None,
195-
dtype: onp.AnyFloat64DType | None = None,
196-
copy: bool = False,
197-
*,
198-
maxprint: int | None = None,
199-
) -> None: ...
200-
@overload # 2-d shape-like, dtype: type[float]
201-
def __init__(
202-
self: csr_array[np.float64, tuple[int, int]],
203-
/,
204-
arg1: ToShape2D,
205-
shape: ToShape2D | None = None,
206-
dtype: onp.AnyFloat64DType | None = None,
207-
copy: bool = False,
208-
*,
209-
maxprint: int | None = None,
210-
) -> None: ...
211189
@overload # 1-d shape-like, dtype: type[complex]
212190
def __init__(
213191
self: csr_array[np.complex128, tuple[int]],

‎scipy-stubs/sparse/_dia.pyi

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -82,13 +82,13 @@ class dia_array(_dia_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G
8282
*,
8383
maxprint: int | None = None,
8484
) -> None: ...
85-
@overload # 2-d shape-like, dtype: None
85+
@overload # 2-d shape-like, dtype: type[float] | None
8686
def __init__(
8787
self: dia_array[np.float64],
8888
/,
8989
arg1: ToShape2D,
9090
shape: ToShape2D | None = None,
91-
dtype: None = None,
91+
dtype: onp.AnyFloat64DType | None = None,
9292
copy: bool = False,
9393
*,
9494
maxprint: int | None = None,
@@ -115,17 +115,6 @@ class dia_array(_dia_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G
115115
*,
116116
maxprint: int | None = None,
117117
) -> None: ...
118-
@overload # 2-d shape-like, dtype: type[float]
119-
def __init__(
120-
self: dia_array[np.float64],
121-
/,
122-
arg1: ToShape2D,
123-
shape: ToShape2D | None = None,
124-
dtype: onp.AnyFloat64DType | None = None,
125-
copy: bool = False,
126-
*,
127-
maxprint: int | None = None,
128-
) -> None: ...
129118
@overload # 2-d shape-like, dtype: type[complex]
130119
def __init__(
131120
self: dia_array[np.complex128],

‎scipy-stubs/sparse/_dok.pyi

Lines changed: 68 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -212,7 +212,7 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT
212212
*,
213213
maxprint: int | None = None,
214214
) -> None: ...
215-
@overload # 1-d shape-like, dtype: None
215+
@overload # 1-d shape-like, dtype: type[float] | None
216216
def __init__(
217217
self: dok_array[np.float64, _1D],
218218
/,
@@ -223,7 +223,7 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT
223223
*,
224224
maxprint: int | None = None,
225225
) -> None: ...
226-
@overload # 2-d shape-like, dtype: None
226+
@overload # 2-d shape-like, dtype: type[float] | None
227227
def __init__(
228228
self: dok_array[np.float64, _2D],
229229
/,
@@ -234,6 +234,72 @@ class dok_array(_dok_base[_ScalarT_co, _ShapeT_co], sparray[_ScalarT_co, _ShapeT
234234
*,
235235
maxprint: int | None = None,
236236
) -> None: ...
237+
@overload # 1-d shape-like, dtype: type[bool]
238+
def __init__(
239+
self: dok_array[np.bool_, _1D],
240+
/,
241+
arg1: ToShape1D,
242+
shape: ToShape1D | None = None,
243+
dtype: onp.AnyBoolDType | None = None,
244+
copy: bool = False,
245+
*,
246+
maxprint: int | None = None,
247+
) -> None: ...
248+
@overload # 2-d shape-like, dtype: type[bool]
249+
def __init__(
250+
self: dok_array[np.bool_, _2D],
251+
/,
252+
arg1: ToShape2D,
253+
shape: ToShape2D | None = None,
254+
dtype: onp.AnyBoolDType | None = None,
255+
copy: bool = False,
256+
*,
257+
maxprint: int | None = None,
258+
) -> None: ...
259+
@overload # 1-d shape-like, dtype: type[int]
260+
def __init__(
261+
self: dok_array[np.int64, _1D],
262+
/,
263+
arg1: ToShape1D,
264+
shape: ToShape1D | None = None,
265+
dtype: onp.AnyIntDType | None = None,
266+
copy: bool = False,
267+
*,
268+
maxprint: int | None = None,
269+
) -> None: ...
270+
@overload # 2-d shape-like, dtype: type[int]
271+
def __init__(
272+
self: dok_array[np.int64, _2D],
273+
/,
274+
arg1: ToShape2D,
275+
shape: ToShape2D | None = None,
276+
dtype: onp.AnyIntDType | None = None,
277+
copy: bool = False,
278+
*,
279+
maxprint: int | None = None,
280+
) -> None: ...
281+
@overload # 1-d shape-like, dtype: type[complex]
282+
def __init__(
283+
self: dok_array[np.complex128, _1D],
284+
/,
285+
arg1: ToShape1D,
286+
shape: ToShape1D | None = None,
287+
dtype: onp.AnyComplex128DType | None = None,
288+
copy: bool = False,
289+
*,
290+
maxprint: int | None = None,
291+
) -> None: ...
292+
@overload # 2-d shape-like, dtype: type[complex]
293+
def __init__(
294+
self: dok_array[np.complex128, _2D],
295+
/,
296+
arg1: ToShape2D,
297+
shape: ToShape2D | None = None,
298+
dtype: onp.AnyComplex128DType | None = None,
299+
copy: bool = False,
300+
*,
301+
maxprint: int | None = None,
302+
) -> None: ...
237303
@overload # 1-d array-like bool, dtype: type[bool] | None
238304
def __init__(
239305
self: dok_array[np.bool_, _1D],

‎scipy-stubs/sparse/_lil.pyi

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -103,13 +103,13 @@ class lil_array(_lil_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G
103103
*,
104104
maxprint: int | None = None,
105105
) -> None: ...
106-
@overload # 2-d shape-like, dtype: None
106+
@overload # 2-d shape-like, dtype: type[float] | None
107107
def __init__(
108108
self: lil_array[np.float64],
109109
/,
110110
arg1: ToShape2D,
111111
shape: ToShape2D | None = None,
112-
dtype: None = None,
112+
dtype: onp.AnyFloat64DType | None = None,
113113
copy: bool = False,
114114
*,
115115
maxprint: int | None = None,
@@ -136,17 +136,6 @@ class lil_array(_lil_base[_ScalarT_co], sparray[_ScalarT_co, tuple[int, int]], G
136136
*,
137137
maxprint: int | None = None,
138138
) -> None: ...
139-
@overload # 2-d shape-like, dtype: type[float]
140-
def __init__(
141-
self: lil_array[np.float64],
142-
/,
143-
arg1: ToShape2D,
144-
shape: ToShape2D | None = None,
145-
dtype: onp.AnyFloat64DType | None = None,
146-
copy: bool = False,
147-
*,
148-
maxprint: int | None = None,
149-
) -> None: ...
150139
@overload # 2-d shape-like, dtype: type[complex]
151140
def __init__(
152141
self: lil_array[np.complex128],

0 commit comments

Comments
 (0)