Skip to content

Commit 6276844

Browse files
jorenhamjorenham
committed
🥅 array constructor workarounds for Any shape overload bugs
1 parent 2cc2f53 commit 6276844

File tree

3 files changed

+208
-84
lines changed

3 files changed

+208
-84
lines changed

‎src/numpy-stubs/@test/static/accept/array_constructors.pyi

Lines changed: 54 additions & 54 deletions
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,10 @@ def func(i: int, j: int, **kwargs: object) -> MyArray[np.float64]: ...
2424
assert_type(np.array(A), _nt.Array[np.float64])
2525
assert_type(np.array(B), _nt.Array1D[np.float64])
2626
assert_type(np.array(D), _nt.Array1D[np.float64 | np.int64])
27-
assert_type(np.array([1, 1.0]), _nt.Array[Any])
28-
assert_type(np.array(deque([1, 2, 3])), _nt.Array[Any])
27+
assert_type(np.array([1, 1.0]), _nt.Array)
28+
assert_type(np.array(deque([1, 2, 3])), _nt.Array)
2929
assert_type(np.array(A, dtype=np.int64), _nt.Array[np.int64])
30-
assert_type(np.array(A, dtype="c16"), _nt.Array[Any])
30+
assert_type(np.array(A, dtype="c16"), _nt.Array)
3131
assert_type(np.array(A, like=A), _nt.Array[np.float64])
3232
assert_type(np.array(A, subok=True), _nt.Array[np.float64])
3333
assert_type(np.array(B, subok=True), MyArray[np.float64])
@@ -36,68 +36,68 @@ assert_type(np.array(B, subok=True, ndmin=1), MyArray[np.float64])
3636

3737
assert_type(np.zeros([1, 5, 6]), _nt.Array[np.float64])
3838
assert_type(np.zeros([1, 5, 6], dtype=np.int64), _nt.Array[np.int64])
39-
assert_type(np.zeros([1, 5, 6], dtype="c16"), _nt.Array[Any])
39+
assert_type(np.zeros([1, 5, 6], dtype="c16"), _nt.Array)
4040

4141
assert_type(np.empty([1, 5, 6]), _nt.Array[np.float64])
4242
assert_type(np.empty([1, 5, 6], dtype=np.int64), _nt.Array[np.int64])
43-
assert_type(np.empty([1, 5, 6], dtype="c16"), _nt.Array[Any])
43+
assert_type(np.empty([1, 5, 6], dtype="c16"), _nt.Array)
4444

4545
assert_type(np.concatenate(A), _nt.Array[np.float64])
46-
assert_type(np.concatenate([A, A]), _nt.Array[Any]) # pyright: ignore[reportAssertTypeFailure] # _nt.Array[np.float64]
47-
assert_type(np.concatenate([[1], A]), _nt.Array[Any])
48-
assert_type(np.concatenate([[1], [1]]), _nt.Array[Any])
46+
assert_type(np.concatenate([A, A]), _nt.Array[np.float64]) # type: ignore[assert-type] # mypy fail
47+
assert_type(np.concatenate([[1], A]), _nt.Array)
48+
assert_type(np.concatenate([[1], [1]]), _nt.Array)
4949
assert_type(np.concatenate((A, A)), _nt.Array[np.float64])
50-
assert_type(np.concatenate(([1], [1])), _nt.Array[Any])
51-
assert_type(np.concatenate([1, 1.0]), _nt.Array[Any])
50+
assert_type(np.concatenate(([1], [1])), _nt.Array)
51+
assert_type(np.concatenate([1, 1.0]), _nt.Array)
5252
assert_type(np.concatenate(A, dtype=np.int64), _nt.Array[np.int64])
53-
assert_type(np.concatenate(A, dtype="c16"), _nt.Array[Any])
53+
assert_type(np.concatenate(A, dtype="c16"), _nt.Array)
5454
assert_type(np.concatenate([1, 1.0], out=A), _nt.Array[np.float64])
5555

5656
assert_type(np.asarray(A), _nt.Array[np.float64])
5757
assert_type(np.asarray(B), _nt.Array1D[np.float64])
58-
assert_type(np.asarray([1, 1.0]), _nt.Array[Any])
58+
assert_type(np.asarray([1, 1.0]), _nt.Array)
5959
assert_type(np.asarray(A, dtype=np.int64), _nt.Array[np.int64])
60-
assert_type(np.asarray(A, dtype="c16"), _nt.Array[Any])
60+
assert_type(np.asarray(A, dtype="c16"), _nt.Array)
6161

6262
assert_type(np.asanyarray(A), _nt.Array[np.float64])
6363
assert_type(np.asanyarray(B), MyArray[np.float64])
64-
assert_type(np.asanyarray([1, 1.0]), _nt.Array[Any])
64+
assert_type(np.asanyarray([1, 1.0]), _nt.Array)
6565
assert_type(np.asanyarray(A, dtype=np.int64), _nt.Array[np.int64])
66-
assert_type(np.asanyarray(A, dtype="c16"), _nt.Array[Any])
66+
assert_type(np.asanyarray(A, dtype="c16"), _nt.Array)
6767

6868
assert_type(np.ascontiguousarray(A), _nt.Array[np.float64])
6969
assert_type(np.ascontiguousarray(B), _nt.Array1D[np.float64])
70-
assert_type(np.ascontiguousarray([1, 1.0]), _nt.Array[Any])
70+
assert_type(np.ascontiguousarray([1, 1.0]), _nt.Array)
7171
assert_type(np.ascontiguousarray(A, dtype=np.int64), _nt.Array[np.int64])
72-
assert_type(np.ascontiguousarray(A, dtype="c16"), _nt.Array[Any])
72+
assert_type(np.ascontiguousarray(A, dtype="c16"), _nt.Array)
7373

7474
assert_type(np.asfortranarray(A), _nt.Array[np.float64])
7575
assert_type(np.asfortranarray(B), _nt.Array1D[np.float64])
76-
assert_type(np.asfortranarray([1, 1.0]), _nt.Array[Any])
76+
assert_type(np.asfortranarray([1, 1.0]), _nt.Array)
7777
assert_type(np.asfortranarray(A, dtype=np.int64), _nt.Array[np.int64])
78-
assert_type(np.asfortranarray(A, dtype="c16"), _nt.Array[Any])
78+
assert_type(np.asfortranarray(A, dtype="c16"), _nt.Array)
7979

8080
assert_type(np.fromstring("1 1 1", sep=" "), _nt.Array[np.float64])
8181
assert_type(np.fromstring(b"1 1 1", sep=" "), _nt.Array[np.float64])
8282
assert_type(np.fromstring("1 1 1", dtype=np.int64, sep=" "), _nt.Array[np.int64])
8383
assert_type(np.fromstring(b"1 1 1", dtype=np.int64, sep=" "), _nt.Array[np.int64])
84-
assert_type(np.fromstring("1 1 1", dtype="c16", sep=" "), _nt.Array[Any])
85-
assert_type(np.fromstring(b"1 1 1", dtype="c16", sep=" "), _nt.Array[Any])
84+
assert_type(np.fromstring("1 1 1", dtype="c16", sep=" "), _nt.Array)
85+
assert_type(np.fromstring(b"1 1 1", dtype="c16", sep=" "), _nt.Array)
8686

8787
assert_type(np.fromfile("test.txt", sep=" "), _nt.Array[np.float64])
8888
assert_type(np.fromfile("test.txt", dtype=np.int64, sep=" "), _nt.Array[np.int64])
89-
assert_type(np.fromfile("test.txt", dtype="c16", sep=" "), _nt.Array[Any])
89+
assert_type(np.fromfile("test.txt", dtype="c16", sep=" "), _nt.Array)
9090
with open("test.txt", encoding="utf-8") as f:
9191
assert_type(np.fromfile(f, sep=" "), _nt.Array[np.float64])
9292
assert_type(np.fromfile(b"test.txt", sep=" "), _nt.Array[np.float64])
9393
assert_type(np.fromfile(Path("test.txt"), sep=" "), _nt.Array[np.float64])
9494

9595
assert_type(np.fromiter("12345", np.float64), _nt.Array[np.float64])
96-
assert_type(np.fromiter("12345", float), _nt.Array[Any])
96+
assert_type(np.fromiter("12345", float), _nt.Array)
9797

9898
assert_type(np.frombuffer(A), _nt.Array[np.float64])
9999
assert_type(np.frombuffer(A, dtype=np.int64), _nt.Array[np.int64])
100-
assert_type(np.frombuffer(A, dtype="c16"), _nt.Array[Any])
100+
assert_type(np.frombuffer(A, dtype="c16"), _nt.Array)
101101

102102
assert_type(np.arange(False, True), _nt.Array1D[np.int_])
103103
assert_type(np.arange(10), _nt.Array1D[np.int_])
@@ -115,7 +115,7 @@ assert_type(np.arange(0, 10, dtype="f8"), _nt.Array1D[np.float64])
115115
assert_type(np.require(A), _nt.Array[np.float64])
116116
assert_type(np.require(B), MyArray[np.float64])
117117
assert_type(np.require(B, requirements=None), MyArray[np.float64])
118-
assert_type(np.require(B, dtype=int), _nt.Array[Any])
118+
assert_type(np.require(B, dtype=int), _nt.Array)
119119
assert_type(np.require(B, requirements="E"), _nt.Array[np.float64])
120120
assert_type(np.require(B, requirements=["ENSUREARRAY"]), _nt.Array[np.float64])
121121
assert_type(np.require(B, requirements={"F", "E"}), _nt.Array[np.float64])
@@ -127,28 +127,28 @@ assert_type(np.require(C), _nt.Array[np.intp])
127127
assert_type(np.linspace(0, 10), _nt.Array[np.floating])
128128
assert_type(np.linspace(0, 10j), _nt.Array[np.inexact])
129129
assert_type(np.linspace(0, 10, dtype=np.int64), _nt.Array[np.int64])
130-
assert_type(np.linspace(0, 10, dtype=int), _nt.Array[Any])
130+
assert_type(np.linspace(0, 10, dtype=int), _nt.Array)
131131
assert_type(np.linspace(0, 10, retstep=True), tuple[_nt.Array[np.floating], np.floating])
132132
assert_type(np.linspace(0j, 10, retstep=True), tuple[_nt.Array[np.inexact], np.inexact])
133133
assert_type(np.linspace(0, 10, retstep=True, dtype=np.int64), tuple[_nt.Array[np.int64], np.int64])
134-
assert_type(np.linspace(0j, 10, retstep=True, dtype=int), tuple[_nt.Array[Any], Any])
134+
assert_type(np.linspace(0j, 10, retstep=True, dtype=int), tuple[_nt.Array, Any])
135135

136136
assert_type(np.logspace(0, 10), _nt.Array[np.floating])
137137
assert_type(np.logspace(0, 10j), _nt.Array[np.inexact])
138138
assert_type(np.logspace(0, 10, dtype=np.int64), _nt.Array[np.int64])
139-
assert_type(np.logspace(0, 10, dtype=int), _nt.Array[Any])
139+
assert_type(np.logspace(0, 10, dtype=int), _nt.Array)
140140

141141
assert_type(np.geomspace(0, 10), _nt.Array[np.floating])
142142
assert_type(np.geomspace(0, 10j), _nt.Array[np.inexact])
143143
assert_type(np.geomspace(0, 10, dtype=np.int64), _nt.Array[np.int64])
144-
assert_type(np.geomspace(0, 10, dtype=int), _nt.Array[Any])
144+
assert_type(np.geomspace(0, 10, dtype=int), _nt.Array)
145145

146146
assert_type(np.empty_like(A), _nt.Array[np.float64])
147147
assert_type(np.empty_like(A, dtype=float), _nt.Array[np.float64])
148-
assert_type(np.empty_like(A, shape=(2, 2)), _nt.Array[np.float64, _nt.Shape2])
148+
assert_type(np.empty_like(A, shape=(2, 2)), _nt.Array2D[np.float64])
149149
assert_type(np.empty_like(A, dtype=np.int64), _nt.Array[np.int64])
150150
assert_type(np.empty_like(A, dtype=np.int64, shape=(2, 2)), _nt.Array[np.int64, _nt.Shape2])
151-
assert_type(np.empty_like(A, dtype="c16"), _nt.Array[Any])
151+
assert_type(np.empty_like(A, dtype="c16"), _nt.Array)
152152
assert_type(np.empty_like(A, dtype="c16", shape=(2, 2)), np.ndarray[_nt.Shape2, np.dtype])
153153
assert_type(np.empty_like(B), MyArray[np.float64])
154154
assert_type(np.empty_like(B, dtype=np.int64), _nt.Array[np.int64])
@@ -171,10 +171,10 @@ assert_type(np.empty_like([[[4j]]]), _nt.Array3D[np.complex128])
171171

172172
assert_type(np.zeros_like(A), _nt.Array[np.float64])
173173
assert_type(np.zeros_like(A, dtype=float), _nt.Array[np.float64])
174-
assert_type(np.zeros_like(A, shape=(2, 2)), _nt.Array[np.float64, _nt.Shape2])
174+
assert_type(np.zeros_like(A, shape=(2, 2)), _nt.Array2D[np.float64])
175175
assert_type(np.zeros_like(A, dtype=np.int64), _nt.Array[np.int64])
176176
assert_type(np.zeros_like(A, dtype=np.int64, shape=(2, 2)), _nt.Array[np.int64, _nt.Shape2])
177-
assert_type(np.zeros_like(A, dtype="c16"), _nt.Array[Any])
177+
assert_type(np.zeros_like(A, dtype="c16"), _nt.Array)
178178
assert_type(np.zeros_like(A, dtype="c16", shape=(2, 2)), np.ndarray[_nt.Shape2, np.dtype])
179179
assert_type(np.zeros_like(B), MyArray[np.float64])
180180
assert_type(np.zeros_like(B, dtype=np.int64), _nt.Array[np.int64])
@@ -197,10 +197,10 @@ assert_type(np.zeros_like([[[4j]]]), _nt.Array3D[np.complex128])
197197

198198
assert_type(np.ones_like(A), _nt.Array[np.float64])
199199
assert_type(np.ones_like(A, dtype=float), _nt.Array[np.float64])
200-
assert_type(np.ones_like(A, shape=(2, 2)), _nt.Array[np.float64, _nt.Shape2])
200+
assert_type(np.ones_like(A, shape=(2, 2)), _nt.Array2D[np.float64])
201201
assert_type(np.ones_like(A, dtype=np.int64), _nt.Array[np.int64])
202202
assert_type(np.ones_like(A, dtype=np.int64, shape=(2, 2)), _nt.Array[np.int64, _nt.Shape2])
203-
assert_type(np.ones_like(A, dtype="c16"), _nt.Array[Any])
203+
assert_type(np.ones_like(A, dtype="c16"), _nt.Array)
204204
assert_type(np.ones_like(A, dtype="c16", shape=(2, 2)), np.ndarray[_nt.Shape2, np.dtype])
205205
assert_type(np.ones_like(B), MyArray[np.float64])
206206
assert_type(np.ones_like(B, dtype=np.int64), _nt.Array[np.int64])
@@ -223,7 +223,7 @@ assert_type(np.ones_like([[[4j]]]), _nt.Array3D[np.complex128])
223223

224224
assert_type(np.full_like(A, i8), _nt.Array[np.float64])
225225
assert_type(np.full_like(C, i8), _nt.Array1D[np.intp])
226-
assert_type(np.full_like(A, i8, dtype=int), _nt.Array[Any])
226+
assert_type(np.full_like(A, i8, dtype=int), _nt.Array)
227227
assert_type(np.full_like(B, i8), MyArray[np.float64])
228228
assert_type(np.full_like(B, i8, dtype=np.int64), _nt.Array[np.int64])
229229

@@ -240,8 +240,8 @@ assert_type(np.ones(_shape_2d), _nt.Array[np.float64, _nt.Shape2])
240240
assert_type(np.ones(_shape_nd), np.ndarray[_nt.Shape, np.dtype[np.float64]])
241241
assert_type(np.ones(_shape_1d, dtype=np.int64), _nt.Array1D[np.int64])
242242
assert_type(np.ones(_shape_like), _nt.Array[np.float64])
243-
assert_type(np.ones(_shape_like, dtype=np.dtypes.StringDType()), np.ndarray[Any, np.dtypes.StringDType])
244-
assert_type(np.ones(_shape_like, dtype=int), _nt.Array[Any])
243+
assert_type(np.ones(_shape_like, dtype=np.dtypes.StringDType()), np.ndarray[_nt.AnyShape, np.dtypes.StringDType])
244+
assert_type(np.ones(_shape_like, dtype=int), _nt.Array)
245245

246246
assert_type(np.full(_size, i8), _nt.Array1D[np.int64])
247247
assert_type(np.full(_shape_2d, i8), _nt.Array[np.int64, _nt.Shape2])
@@ -265,12 +265,12 @@ assert_type(np.identity(10, dtype=np.int64), _nt.Array2D[np.int64])
265265
assert_type(np.identity(10, dtype=int), _nt.Array2D)
266266

267267
assert_type(np.atleast_1d(A), _nt.Array[np.float64])
268-
assert_type(np.atleast_1d(C), _nt.Array[Any])
268+
assert_type(np.atleast_1d(C), _nt.Array)
269269
assert_type(np.atleast_1d(A, A), tuple[_nt.Array[np.float64], _nt.Array[np.float64]])
270-
assert_type(np.atleast_1d(A, C), tuple[_nt.Array[Any], _nt.Array[Any]])
271-
assert_type(np.atleast_1d(C, C), tuple[_nt.Array[Any], _nt.Array[Any]])
270+
assert_type(np.atleast_1d(A, C), tuple[_nt.Array, _nt.Array])
271+
assert_type(np.atleast_1d(C, C), tuple[_nt.Array, _nt.Array])
272272
assert_type(np.atleast_1d(A, A, A), tuple[_nt.Array[np.float64], ...])
273-
assert_type(np.atleast_1d(C, C, C), tuple[_nt.Array[Any], ...])
273+
assert_type(np.atleast_1d(C, C, C), tuple[_nt.Array, ...])
274274

275275
assert_type(np.atleast_2d(A), _nt.Array[np.float64])
276276
assert_type(np.atleast_2d(A, A), tuple[_nt.Array[np.float64], _nt.Array[np.float64]])
@@ -280,28 +280,28 @@ assert_type(np.atleast_3d(A), _nt.Array[np.float64])
280280
assert_type(np.atleast_3d(A, A), tuple[_nt.Array[np.float64], _nt.Array[np.float64]])
281281
assert_type(np.atleast_3d(A, A, A), tuple[_nt.Array[np.float64], ...])
282282

283-
assert_type(np.vstack([A, A]), _nt.Array[np.float64])
283+
assert_type(np.vstack([A, A]), _nt.Array[np.float64]) # type: ignore[assert-type] # mypy fail
284284
assert_type(np.vstack([A, A], dtype=np.float32), _nt.Array[np.float32])
285-
assert_type(np.vstack([A, C]), _nt.Array[Any])
286-
assert_type(np.vstack([C, C]), _nt.Array[Any])
285+
assert_type(np.vstack([A, C]), _nt.Array)
286+
assert_type(np.vstack([C, C]), _nt.Array)
287287

288-
assert_type(np.hstack([A, A]), _nt.Array[np.float64])
288+
assert_type(np.hstack([A, A]), _nt.Array[np.float64]) # type: ignore[assert-type] # mypy fail
289289
assert_type(np.hstack([A, A], dtype=np.float32), _nt.Array[np.float32])
290290

291-
assert_type(np.stack([A, A]), _nt.Array[np.float64])
291+
assert_type(np.stack([A, A]), _nt.Array[np.float64]) # type: ignore[assert-type] # mypy fail
292292
assert_type(np.stack([A, A], dtype=np.float32), _nt.Array[np.float32])
293-
assert_type(np.stack([A, C]), _nt.Array[Any])
294-
assert_type(np.stack([C, C]), _nt.Array[Any])
295-
assert_type(np.stack([A, A], axis=0), _nt.Array[np.float64])
293+
assert_type(np.stack([A, C]), _nt.Array)
294+
assert_type(np.stack([C, C]), _nt.Array)
295+
assert_type(np.stack([A, A], axis=0), _nt.Array[np.float64]) # type: ignore[assert-type] # mypy fail
296296
assert_type(np.stack([A, A], out=B), MyArray[np.float64])
297297

298-
assert_type(np.block([[A, A], [A, A]]), _nt.Array[Any]) # pyright: ignore[reportAssertTypeFailure] # _nt.Array[np.float64]
299-
assert_type(np.block(C), _nt.Array[Any])
298+
assert_type(np.block([[A, A], [A, A]]), _nt.Array) # pyright: ignore[reportAssertTypeFailure] # _nt.Array[np.float64]
299+
assert_type(np.block(C), _nt.Array)
300300

301301
if sys.version_info >= (3, 12):
302302
from collections.abc import Buffer
303303

304-
def create_array(obj: npt.ArrayLike) -> _nt.Array[Any]: ...
304+
def create_array(obj: npt.ArrayLike) -> _nt.Array: ...
305305

306306
buffer: Buffer
307-
assert_type(create_array(buffer), _nt.Array[Any])
307+
assert_type(create_array(buffer), _nt.Array)

‎src/numpy-stubs/_core/_multiarray_umath.pyi

Lines changed: 49 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -599,6 +599,16 @@ def empty_like(
599599
*,
600600
device: _Device | None = None,
601601
) -> np.ndarray[_ShapeT, _DTypeT]: ...
602+
@overload # workaround for microsoft/pyright#10232
603+
def empty_like(
604+
prototype: _nt._ToArray_nnd[np.bool_],
605+
dtype: _nt.ToDTypeBool | None = None,
606+
order: _OrderKACF = "K",
607+
subok: bool = True,
608+
shape: tuple[()] | None = None,
609+
*,
610+
device: _Device | None = None,
611+
) -> _nt.Array[np.bool_]: ...
602612
@overload # bool 0d array-like
603613
def empty_like( # type: ignore[overload-overlap]
604614
prototype: _nt.ToBool_0d,
@@ -639,6 +649,16 @@ def empty_like(
639649
*,
640650
device: _Device | None = None,
641651
) -> _nt.Array3D[np.bool_]: ...
652+
@overload # workaround for microsoft/pyright#10232
653+
def empty_like(
654+
prototype: _nt._ToArray_nnd[np.intp],
655+
dtype: _nt.ToDTypeInt64 | None = None,
656+
order: _OrderKACF = "K",
657+
subok: bool = True,
658+
shape: tuple[()] | None = None,
659+
*,
660+
device: _Device | None = None,
661+
) -> _nt.Array[np.intp]: ...
642662
@overload # int 0d array-like
643663
def empty_like( # type: ignore[overload-overlap]
644664
prototype: _nt.ToInt_0d,
@@ -679,6 +699,16 @@ def empty_like(
679699
*,
680700
device: _Device | None = None,
681701
) -> _nt.Array3D[np.intp]: ...
702+
@overload # workaround for microsoft/pyright#10232
703+
def empty_like(
704+
prototype: _nt._ToArray_nnd[np.float64],
705+
dtype: _nt.ToDTypeFloat64 | None = None,
706+
order: _OrderKACF = "K",
707+
subok: bool = True,
708+
shape: tuple[()] | None = None,
709+
*,
710+
device: _Device | None = None,
711+
) -> _nt.Array[np.float64]: ...
682712
@overload # float 0d array-like
683713
def empty_like( # type: ignore[overload-overlap]
684714
prototype: _nt.ToFloat64_0d,
@@ -729,6 +759,16 @@ def empty_like( # type: ignore[overload-overlap]
729759
*,
730760
device: _Device | None = None,
731761
) -> _nt.Array0D[np.complex128]: ...
762+
@overload # workaround for microsoft/pyright#10232
763+
def empty_like(
764+
prototype: _nt._ToArray_nnd[np.complex128],
765+
dtype: _nt.ToDTypeComplex128 | None = None,
766+
order: _OrderKACF = "K",
767+
subok: bool = True,
768+
shape: tuple[()] | None = None,
769+
*,
770+
device: _Device | None = None,
771+
) -> _nt.Array[np.complex128]: ...
732772
@overload # complex 1d array-like
733773
def empty_like(
734774
prototype: _nt.ToComplex128_1ds,
@@ -770,7 +810,7 @@ def empty_like( # type: ignore[overload-overlap]
770810
device: _Device | None = None,
771811
) -> _nt.Array[_ScalarT, _AnyShapeT]: ...
772812
@overload # array-like with known scalar-type, unknown shape
773-
def empty_like( # type: ignore[overload-overlap]
813+
def empty_like(
774814
prototype: _ArrayLike[_ScalarT],
775815
dtype: np.dtype[_ScalarT] | None = None,
776816
order: _OrderKACF = "K",
@@ -820,7 +860,7 @@ def empty_like(
820860
device: _Device | None = None,
821861
) -> _nt.Array[_ScalarT]: ...
822862
@overload # bool array-like
823-
def empty_like( # type: ignore[overload-overlap]
863+
def empty_like(
824864
prototype: _nt.ToBool_nd,
825865
dtype: _nt.ToDTypeBool | None = None,
826866
order: _OrderKACF = "K",
@@ -830,7 +870,7 @@ def empty_like( # type: ignore[overload-overlap]
830870
device: _Device | None = None,
831871
) -> _nt.Array[np.bool_]: ...
832872
@overload # int array-like
833-
def empty_like( # type: ignore[overload-overlap]
873+
def empty_like(
834874
prototype: _nt.ToInt_nd,
835875
dtype: _nt.ToDTypeInt64 | None = None,
836876
order: _OrderKACF = "K",
@@ -840,7 +880,7 @@ def empty_like( # type: ignore[overload-overlap]
840880
device: _Device | None = None,
841881
) -> _nt.Array[np.intp]: ...
842882
@overload # float array-like
843-
def empty_like( # type: ignore[overload-overlap]
883+
def empty_like(
844884
prototype: _nt.ToFloat64_nd,
845885
dtype: _nt.ToDTypeFloat64 | None = None,
846886
order: _OrderKACF = "K",
@@ -850,7 +890,7 @@ def empty_like( # type: ignore[overload-overlap]
850890
device: _Device | None = None,
851891
) -> _nt.Array[np.float64]: ...
852892
@overload # complex array-like
853-
def empty_like( # type: ignore[overload-overlap]
893+
def empty_like(
854894
prototype: _nt.ToComplex128_nd,
855895
dtype: _nt.ToDTypeComplex128 | None = None,
856896
order: _OrderKACF = "K",
@@ -932,7 +972,7 @@ def array(
932972
**kwargs: Unpack[_KwargsCL],
933973
) -> _nt.Array[_ScalarT]: ...
934974
@overload
935-
def array( # type: ignore[overload-overlap]
975+
def array(
936976
object: _ScalarLike_co,
937977
dtype: _DTypeLike[_ScalarT],
938978
*,
@@ -942,7 +982,7 @@ def array( # type: ignore[overload-overlap]
942982
**kwargs: Unpack[_KwargsCL],
943983
) -> _nt.Array0D[_ScalarT]: ...
944984
@overload
945-
def array( # type: ignore[overload-overlap]
985+
def array(
946986
object: _ScalarLike_co,
947987
dtype: npt.DTypeLike | None,
948988
*,
@@ -985,11 +1025,11 @@ def asarray(
9851025
a: _ArrayLike[_ScalarT], dtype: None = None, order: _OrderKACF = None, **kwargs: Unpack[_KwargsDCL]
9861026
) -> _nt.Array[_ScalarT]: ...
9871027
@overload
988-
def asarray( # type: ignore[overload-overlap]
1028+
def asarray(
9891029
a: _ScalarLike_co, dtype: _DTypeLike[_ScalarT], order: _OrderKACF = None, **kwargs: Unpack[_KwargsDCL]
9901030
) -> _nt.Array0D[_ScalarT]: ...
9911031
@overload
992-
def asarray( # type: ignore[overload-overlap]
1032+
def asarray(
9931033
a: _ScalarLike_co, dtype: npt.DTypeLike | None = None, order: _OrderKACF = None, **kwargs: Unpack[_KwargsDCL]
9941034
) -> _nt.Array0D[Incomplete]: ...
9951035
@overload

0 commit comments

Comments
 (0)