Skip to content

Commit ce88d83

Browse files
authored
pandas: fix various arguments for Series.iloc and DataFrame.loc (#82)
1 parent 36b8b1e commit ce88d83

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

pandas/core/frame.pyi

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,8 @@ class _iLocIndexerFrame(_iLocIndexer):
5454
) -> None: ...
5555

5656
class _LocIndexerFrame(_LocIndexer):
57+
@overload
58+
def __getitem__(self, idx: Union[Union[IndexType, MaskType], Tuple[Union[IndexType, MaskType, slice], Union[List[StrLike], slice]]],) -> DataFrame: ...
5759
@overload
5860
def __getitem__(self, idx: Tuple[StrLike, StrLike],) -> Scalar: ...
5961
@overload
@@ -63,8 +65,6 @@ class _LocIndexerFrame(_LocIndexer):
6365
@overload
6466
def __getitem__(self, idx: Tuple[Union[IndexType, MaskType], StrLike],) -> Series[Dtype]: ...
6567
@overload
66-
def __getitem__(self, idx: Union[Union[IndexType, MaskType], Tuple[Union[IndexType, MaskType], Union[List[StrLike], slice]]],) -> DataFrame: ...
67-
@overload
6868
def __getitem__(self, idx: Index) -> DataFrame: ...
6969

7070
def __setitem__(

pandas/core/series.pyi

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ class _iLocIndexerSeries(_iLocIndexer, Generic[S1]):
2727
@overload
2828
def __getitem__(self, idx: int) -> S1: ...
2929
@overload
30-
def __getitem__(self, idx: Index) -> Series[S1]: ...
30+
def __getitem__(self, idx: Union[Index, slice]) -> Series[S1]: ...
3131
# set item
3232
@overload
3333
def __setitem__(self, idx: int, value: S1) -> None: ...
@@ -37,7 +37,7 @@ class _iLocIndexerSeries(_iLocIndexer, Generic[S1]):
3737

3838
class _LocIndexerSeries(_LocIndexer, Generic[S1]):
3939
@overload
40-
def __getitem__(self, idx: Union[MaskType, Sequence[str]],) -> Series[S1]: ...
40+
def __getitem__(self, idx: Union[MaskType, Index, Sequence[str], slice],) -> Series[S1]: ...
4141
@overload
4242
def __getitem__(self, idx: Union[int, str],) -> S1: ...
4343
@overload

0 commit comments

Comments
 (0)