Skip to content

Commit cd48b3d

Browse files
authored
✨ improved np.arange signature, and _core._multiarray_umath cleanup (#547)
1 parent 94c4fb1 commit cd48b3d

File tree

2 files changed

+127
-242
lines changed

2 files changed

+127
-242
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -102,15 +102,15 @@ assert_type(np.frombuffer(A, dtype="c16"), _nt.Array[Any])
102102
assert_type(np.arange(False, True), _nt.Array1D[np.int_])
103103
assert_type(np.arange(10), _nt.Array1D[np.int_])
104104
assert_type(np.arange(0, 10, step=2), _nt.Array1D[np.int_])
105-
assert_type(np.arange(10.0), _nt.Array1D[np.float64 | np.int_])
106-
assert_type(np.arange(start=0, stop=10.0), _nt.Array1D[np.float64 | np.int_])
105+
assert_type(np.arange(10.0), _nt.Array1D[np.float64])
106+
assert_type(np.arange(start=0, stop=10.0), _nt.Array1D[np.float64])
107107
assert_type(np.arange(np.timedelta64(0)), _nt.Array1D[np.timedelta64])
108108
assert_type(np.arange(0, np.timedelta64(10)), _nt.Array1D[np.timedelta64])
109109
assert_type(np.arange(np.datetime64("0"), np.datetime64("10")), _nt.Array1D[np.datetime64])
110110
assert_type(np.arange(10, dtype=np.float64), _nt.Array1D[np.float64])
111111
assert_type(np.arange(0, 10, step=2, dtype=np.int16), _nt.Array1D[np.int16])
112112
assert_type(np.arange(10, dtype=int), _nt.Array1D[np.int_])
113-
assert_type(np.arange(0, 10, dtype="f8"), _nt.Array1D)
113+
assert_type(np.arange(0, 10, dtype="f8"), _nt.Array1D[np.float64])
114114

115115
assert_type(np.require(A), _nt.Array[np.float64])
116116
assert_type(np.require(B), MyArray[np.float64])

0 commit comments

Comments
 (0)