10000 Merge pull request #28905 from MarcoGorelli/ma-transpose · numpy/numpy@3418372 · GitHub
[go: up one dir, main page]
Skip to content

Commit 3418372

Browse files
jorenhamjorenham
authored
Merge pull request #28905 from MarcoGorelli/ma-transpose
2 parents 9baed0e + 01f5c86 commit 3418372

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

‎numpy/ma/core.pyi

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -887,9 +887,6 @@ class MaskedArray(ndarray[_ShapeT_co, _DTypeT_co]):
887887
/
888888
) -> MaskedArray[_Shape, _DTypeT_co]: ...
889889

890-
T: Any
891-
transpose: Any
892-
893890
#
894891
def toflex(self) -> Incomplete: ...
895892
def torecords(self) -> Incomplete: ...

‎numpy/typing/tests/data/reveal/ma.pyi

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -360,5 +360,12 @@ assert_type(MAR_b.shrink_mask(), MaskedArray[np.bool_])
360360
assert_type(MAR_i8.hardmask, bool)
361361
assert_type(MAR_i8.sharedmask, bool)
362362

363+
assert_type(MAR_b.transpose(), MaskedArray[np.bool])
364+
assert_type(MAR_2d_f4.transpose(), np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])
365+
assert_type(MAR_2d_f4.transpose(1, 0), np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])
366+
assert_type(MAR_2d_f4.transpose((1, 0)), np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])
367+
assert_type(MAR_b.T, MaskedArray[np.bool])
368+
assert_type(MAR_2d_f4.T, np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])
369+
363370
assert_type(MAR_2d_f4.nonzero(), tuple[_Array1D[np.intp], *tuple[_Array1D[np.intp], ...]])
364371
assert_type(MAR_2d_f4.nonzero()[0], _Array1D[np.intp])

0 commit comments

Comments
 (0)
0