diff --git a/pandas-stubs/core/series.pyi b/pandas-stubs/core/series.pyi index fc45ee5a..ce5a9654 100644 --- a/pandas-stubs/core/series.pyi +++ b/pandas-stubs/core/series.pyi @@ -965,6 +965,12 @@ class Series(IndexOpsMixin[S1], NDFrame): na_action: None = ..., ) -> Series[S2]: ... @overload + def map( + self, + arg: Callable[[Any], Any] | Mapping[Any, Any] | UnknownSeries, + na_action: Literal["ignore"] | None = ..., + ) -> UnknownSeries: ... + @overload def aggregate( self: Series[int], func: Literal["mean"], diff --git a/tests/test_series.py b/tests/test_series.py index ae1b7f3f..3b3604c0 100644 --- a/tests/test_series.py +++ b/tests/test_series.py @@ -3390,6 +3390,12 @@ def callable(x: int) -> str: assert_type(s.map(series, na_action="ignore"), "pd.Series[str]"), pd.Series, str ) + unknown_series = pd.Series([1, 0, None]) + check( + assert_type(unknown_series.map({1: True, 0: False, None: None}), "pd.Series"), + pd.Series, + ) + def test_map_na() -> None: s: pd.Series[int] = pd.Series([1, pd.NA, 3])