Skip to content

Commit 233bdcc

Browse files
GHXXX Clean up from stubtest
1 parent 69cf85e commit 233bdcc

File tree

7 files changed

+34
-20
lines changed

7 files changed

+34
-20
lines changed

pandas-stubs/core/frame.pyi

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1990,15 +1990,13 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
19901990
self,
19911991
axis: Axis | None = ...,
19921992
skipna: _bool | None = ...,
1993-
level: None = ...,
19941993
numeric_only: _bool = ...,
19951994
**kwargs: Any,
19961995
) -> Series: ...
19971996
def kurtosis(
19981997
self,
19991998
axis: Axis | None = ...,
20001999
skipna: _bool | None = ...,
2001-
level: None = ...,
20022000
numeric_only: _bool = ...,
20032001
**kwargs: Any,
20042002
) -> Series: ...
@@ -2042,15 +2040,13 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
20422040
self,
20432041
axis: Axis | None = ...,
20442042
skipna: _bool | None = ...,
2045-
level: None = ...,
20462043
numeric_only: _bool = ...,
20472044
**kwargs: Any,
20482045
) -> Series: ...
20492046
def mean(
20502047
self,
20512048
axis: Axis | None = ...,
20522049
skipna: _bool | None = ...,
2053-
level: None = ...,
20542050
numeric_only: _bool = ...,
20552051
**kwargs: Any,
20562052
) -> Series: ...

pandas-stubs/core/indexes/base.pyi

Lines changed: 16 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,6 @@ from pandas._typing import (
4848
DropKeep,
4949
DtypeArg,
5050
DtypeObj,
51-
FillnaOptions,
5251
HashableT,
5352
Label,
5453
Level,
@@ -313,7 +312,13 @@ class Index(IndexOpsMixin[S1]):
313312
def rename(self, name, inplace: Literal[True]) -> None: ...
314313
@property
315314
def nlevels(self) -> int: ...
316-
def sortlevel(self, level=..., ascending: bool = ..., sort_remaining=...): ...
315+
def sortlevel(
316+
self,
317+
level=...,
318+
ascending: bool = ...,
319+
sort_remaining=...,
320+
na_position: Literal["first", "last"] = ...,
321+
): ...
317322
def get_level_values(self, level: int | _str) -> Index: ...
318323
def droplevel(self, level: Level | list[Level] = ...): ...
319324
@property
@@ -353,12 +358,7 @@ class Index(IndexOpsMixin[S1]):
353358
def symmetric_difference(
354359
self, other: list[S1] | Self, result_name: Hashable = ..., sort=...
355360
) -> Self: ...
356-
def get_loc(
357-
self,
358-
key: Label,
359-
method: FillnaOptions | Literal["nearest"] | None = ...,
360-
tolerance=...,
361-
) -> int | slice | np_ndarray_bool: ...
361+
def get_loc(self, key: Label) -> int | slice | np_ndarray_bool: ...
362362
def get_indexer(
363363
self, target, method: ReindexMethod | None = ..., limit=..., tolerance=...
364364
): ...
@@ -400,7 +400,14 @@ class Index(IndexOpsMixin[S1]):
400400
def identical(self, other) -> bool: ...
401401
def asof(self, label): ...
402402
def asof_locs(self, where, mask): ...
403-
def sort_values(self, return_indexer: bool = ..., ascending: bool = ...): ...
403+
def sort_values(
404+
self,
405+
*,
406+
return_indexer: bool = ...,
407+
ascending: bool = ...,
408+
na_position: Literal["first", "last"] = ...,
409+
key: Callable | None = None,
410+
): ...
404411
def sort(self, *args, **kwargs) -> None: ...
405412
def argsort(self, *args, **kwargs): ...
406413
def get_indexer_non_unique(self, target): ...

pandas-stubs/core/indexes/category.pyi

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,9 @@ class CategoricalIndex(ExtensionIndex[S1], accessor.PandasDelegate):
3333
@property
3434
def values(self): ...
3535
def __contains__(self, key) -> bool: ...
36-
def __array__(self, dtype=...) -> np.ndarray: ...
36+
def __array__(
37+
self, dtype: DtypeArg = ..., copy: bool | None = ...
38+
) -> np.ndarray: ...
3739
def astype(self, dtype: DtypeArg, copy: bool = ...) -> Index: ...
3840
def fillna(self, value=...): ...
3941
@property

pandas-stubs/core/indexes/multi.pyi

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ from collections.abc import (
55
Sequence,
66
)
77
from typing import (
8+
Literal,
89
overload,
910
)
1011

@@ -140,7 +141,11 @@ class MultiIndex(Index):
140141
def swaplevel(self, i: int = ..., j: int = ...): ...
141142
def reorder_levels(self, order): ...
142143
def sortlevel(
143-
self, level: int = ..., ascending: bool = ..., sort_remaining: bool = ...
144+
self,
145+
level: int = ...,
146+
ascending: bool = ...,
147+
sort_remaining: bool = ...,
148+
na_position: Literal["first", "last"] = ...,
144149
): ...
145150
def get_indexer(self, target, method=..., limit=..., tolerance=...): ...
146151
def get_indexer_non_unique(self, target): ...
Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +0,0 @@
1-
from pandas.core.interchange.dataframe_protocol import DataFrame as DataFrame
2-
from pandas.core.interchange.from_dataframe import from_dataframe as from_dataframe

pandas-stubs/core/resample.pyi

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ _SeriesGroupByFuncArgs: TypeAlias = (
6262
)
6363

6464
class Resampler(BaseGroupBy[NDFrameT]):
65-
grouper: BinGrouper # pyright: ignore[reportIncompatibleVariableOverride] # pyrefly: ignore # variance incompatibility
65+
_grouper: BinGrouper
6666
binner: DatetimeIndex | TimedeltaIndex | PeriodIndex
6767
exclusions: frozenset[Hashable]
6868
ax: Index
@@ -123,8 +123,12 @@ class Resampler(BaseGroupBy[NDFrameT]):
123123
def prod(self, numeric_only: bool = ..., min_count: int = ...) -> NDFrameT: ...
124124
def min(self, numeric_only: bool = ..., min_count: int = ...) -> NDFrameT: ...
125125
def max(self, numeric_only: bool = ..., min_count: int = ...) -> NDFrameT: ...
126-
def first(self, numeric_only: bool = ..., min_count: int = ...) -> NDFrameT: ...
127-
def last(self, numeric_only: bool = ..., min_count: int = ...) -> NDFrameT: ...
126+
def first(
127+
self, numeric_only: bool = ..., min_count: int = ..., skipna: bool = ...
128+
) -> NDFrameT: ...
129+
def last(
130+
self, numeric_only: bool = ..., min_count: int = ..., skipna: bool = ...
131+
) -> NDFrameT: ...
128132
def median(self, numeric_only: bool = ...) -> NDFrameT: ...
129133
def mean(self, numeric_only: bool = ...) -> NDFrameT: ...
130134
def std(self, ddof: int = ..., numeric_only: bool = ...) -> NDFrameT: ...

tests/test_resampler.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,9 @@ def test_agg_funcs() -> None:
5353
check(assert_type(DF.resample("ME").min(), DataFrame), DataFrame)
5454
check(assert_type(DF.resample("ME").max(), DataFrame), DataFrame)
5555
check(assert_type(DF.resample("ME").first(), DataFrame), DataFrame)
56+
check(assert_type(DF.resample("ME").first(skipna=False), DataFrame), DataFrame)
5657
check(assert_type(DF.resample("ME").last(), DataFrame), DataFrame)
58+
check(assert_type(DF.resample("ME").last(skipna=False), DataFrame), DataFrame)
5759
check(assert_type(DF.resample("ME").mean(), DataFrame), DataFrame)
5860
check(assert_type(DF.resample("ME").sum(), DataFrame), DataFrame)
5961
check(assert_type(DF.resample("ME").median(), DataFrame), DataFrame)

0 commit comments

Comments
 (0)