Skip to content

Commit f2ba874

Browse files
committed
use Sequence[Sequence[Scalar]]
1 parent 02f940a commit f2ba874

File tree

2 files changed

+32
-10
lines changed

2 files changed

+32
-10
lines changed

pandas-stubs/core/frame.pyi

Lines changed: 30 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -292,7 +292,8 @@ class _LocIndexerFrame(_LocIndexer, Generic[_T]):
292292
| ArrayLike
293293
| IndexOpsMixin
294294
| DataFrame
295-
| list
295+
| Sequence[Scalar]
296+
| Sequence[Sequence[Scalar]]
296297
| Mapping[Hashable, Scalar | NAType | NaTType]
297298
| None
298299
),
@@ -301,7 +302,17 @@ class _LocIndexerFrame(_LocIndexer, Generic[_T]):
301302
def __setitem__(
302303
self,
303304
idx: tuple[_IndexSliceTuple, Hashable],
304-
value: Scalar | NAType | NaTType | ArrayLike | Series | list | dict | None,
305+
value: (
306+
Scalar
307+
| NAType
308+
| NaTType
309+
| ArrayLike
310+
| Series
311+
| Sequence[Scalar]
312+
| Sequence[Sequence[Scalar]]
313+
| Mapping[Hashable, Scalar | NAType | NaTType]
314+
| None
315+
),
305316
) -> None: ...
306317

307318
class _iAtIndexerFrame(_iAtIndexer):
@@ -336,7 +347,8 @@ class _AtIndexerFrame(_AtIndexer):
336347
| ArrayLike
337348
| IndexOpsMixin
338349
| DataFrame
339-
| list
350+
| Sequence[Scalar]
351+
| Sequence[Sequence[Scalar]]
340352
| Mapping[Hashable, Scalar | NAType | NaTType]
341353
| None
342354
),
@@ -824,7 +836,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
824836
| IndexOpsMixin
825837
| Sequence[Scalar]
826838
| DataFrame
827-
| np.ndarray
839+
| np_ndarray
828840
| NAType
829841
| NaTType
830842
| Mapping[Hashable, Scalar | NAType | NaTType]
@@ -845,8 +857,8 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
845857
| ArrayLike
846858
| IndexOpsMixin
847859
| Sequence[Scalar]
860+
| Sequence[Sequence[Scalar]]
848861
| DataFrame
849-
| list
850862
| Mapping[Hashable, Scalar | NAType | NaTType]
851863
| None
852864
),
@@ -862,7 +874,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
862874
| ArrayLike
863875
| IndexOpsMixin
864876
| Sequence[Scalar]
865-
| dict
877+
| Mapping[Hashable, Scalar | NAType | NaTType]
866878
| None
867879
),
868880
) -> None: ...
@@ -872,8 +884,18 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
872884
def __setitem__(
873885
self,
874886
idx: IndexOpsMixin | DataFrame,
875-
value: Scalar | NAType | NaTType | ArrayLike | Series | list | dict | None,
876-
): ...
887+
value: (
888+
Scalar
889+
| NAType
890+
| NaTType
891+
| ArrayLike
892+
| Series
893+
| Sequence[Scalar]
894+
| Sequence[Sequence[Scalar]]
895+
| Mapping[Hashable, Scalar | NAType | NaTType]
896+
| None
897+
),
898+
) -> None: ...
877899
@overload
878900
def query(
879901
self,

pandas-stubs/core/indexing.pyi

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from collections.abc import Sequence
12
from typing import (
23
TypeAlias,
34
TypeVar,
@@ -9,11 +10,10 @@ from pandas._libs.indexing import _NDFrameIndexerBase
910
from pandas._typing import (
1011
MaskType,
1112
Scalar,
12-
ScalarT,
1313
)
1414

1515
_IndexSliceTuple: TypeAlias = tuple[
16-
IndexOpsMixin | MaskType | Scalar | list[ScalarT] | slice | tuple[Scalar, ...], ...
16+
IndexOpsMixin | MaskType | Scalar | Sequence[Scalar] | slice, ...
1717
]
1818

1919
_IndexSliceUnion: TypeAlias = slice | _IndexSliceTuple

0 commit comments

Comments
 (0)