Skip to content
Open
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions pandas-stubs/core/series.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -1107,7 +1107,7 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
def map(
self,
arg: Callable[[S1], S2 | NAType] | Mapping[S1, S2] | Series[S2],
na_action: Literal["ignore"] = ...,
na_action: Literal["ignore"],
) -> Series[S2]: ...
@overload
def map(
Expand All @@ -1119,7 +1119,7 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
def map(
self,
arg: Callable[[Any], Any] | Mapping[Any, Any] | Series,
na_action: Literal["ignore"] | None = ...,
na_action: Literal["ignore"] | None = None,
) -> Series: ...
@overload
def aggregate(
Expand Down
2 changes: 2 additions & 0 deletions tests/series/test_series.py
Original file line number Diff line number Diff line change
Expand Up @@ -3620,6 +3620,8 @@ def callable(x: int | NAType) -> str | NAType:
check(
assert_type(s.map(callable, na_action=None), "pd.Series[str]"), pd.Series, str
)
# na_action defaults to None
check(assert_type(s.map(callable), "pd.Series[str]"), pd.Series, str)

series = pd.Series(["a", "b", "c"])
check(assert_type(s.map(series, na_action=None), "pd.Series[str]"), pd.Series, str)
Expand Down