Skip to content

Commit 0f14e32

Browse files
authored
Add some missing types to BaseOffset and ExtensionArray (#1330)
improve typing of BaseOffset and ExtensionArray
1 parent 3aa4044 commit 0f14e32

File tree

2 files changed

+7
-9
lines changed

2 files changed

+7
-9
lines changed

pandas-stubs/_libs/tslibs/offsets.pyi

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -29,11 +29,11 @@ class ApplyTypeError(TypeError): ...
2929
class BaseOffset:
3030
n: int
3131
def __init__(self, n: int = ..., normalize: bool = ...) -> None: ...
32-
def __eq__(self, other) -> bool: ...
33-
def __ne__(self, other) -> bool: ...
32+
def __eq__(self, other: object) -> bool: ...
33+
def __ne__(self, other: object) -> bool: ...
3434
def __hash__(self) -> int: ...
3535
@property
36-
def kwds(self) -> dict: ...
36+
def kwds(self) -> dict[str, Any]: ...
3737
@property
3838
def base(self) -> BaseOffset: ...
3939
@overload
@@ -67,7 +67,6 @@ class BaseOffset:
6767
def __rsub__(self, other: BaseOffset) -> Self: ...
6868
@overload
6969
def __rsub__(self, other: _TimedeltaT) -> _TimedeltaT: ...
70-
def __call__(self, other): ...
7170
@overload
7271
def __mul__(self, other: np.ndarray) -> np.ndarray: ...
7372
@overload
@@ -87,8 +86,6 @@ class BaseOffset:
8786
def rollback(self, dt: datetime) -> datetime: ...
8887
def rollforward(self, dt: datetime) -> datetime: ...
8988
def is_on_offset(self, dt: datetime) -> bool: ...
90-
def __setstate__(self, state) -> None: ...
91-
def __getstate__(self): ...
9289
@property
9390
def nanos(self) -> int: ...
9491

pandas-stubs/core/arrays/base.pyi

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from collections.abc import Iterator
12
from typing import (
23
Any,
34
overload,
@@ -23,16 +24,16 @@ class ExtensionArray:
2324
def __getitem__(self, item: ScalarIndexer) -> Any: ...
2425
@overload
2526
def __getitem__(self, item: SequenceIndexer) -> Self: ...
26-
def __setitem__(self, key: int | slice | np.ndarray, value) -> None: ...
27+
def __setitem__(self, key: int | slice | np.ndarray, value: Any) -> None: ...
2728
def __len__(self) -> int: ...
28-
def __iter__(self): ...
29+
def __iter__(self) -> Iterator[Any]: ...
2930
def __contains__(self, item: object) -> bool | np.bool_: ...
3031
def to_numpy(
3132
self,
3233
dtype: npt.DTypeLike | None = ...,
3334
copy: bool = False,
3435
na_value: Scalar = ...,
35-
) -> np_1darray: ...
36+
) -> np_1darray[Any]: ...
3637
@property
3738
def dtype(self) -> ExtensionDtype: ...
3839
@property

0 commit comments

Comments
 (0)