From 9195c233676b9c3440ab8115a04161cd61c9fbbb Mon Sep 17 00:00:00 2001 From: Olga Matoula Date: Tue, 13 Apr 2021 00:05:14 +0100 Subject: [PATCH] TYP: Add type to arg of series/drop-duplicates --- pandas/core/series.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/pandas/core/series.py b/pandas/core/series.py index 5ba68aaa5c16d..93e3fbf769118 100644 --- a/pandas/core/series.py +++ b/pandas/core/series.py @@ -2004,11 +2004,13 @@ def unique(self) -> ArrayLike: return super().unique() @overload - def drop_duplicates(self, keep=..., inplace: Literal[False] = ...) -> Series: + def drop_duplicates( + self, keep: str | bool = ..., inplace: Literal[False] = ... + ) -> Series: ... @overload - def drop_duplicates(self, keep, inplace: Literal[True]) -> None: + def drop_duplicates(self, keep: str | bool, inplace: Literal[True]) -> None: ... @overload @@ -2016,10 +2018,14 @@ def drop_duplicates(self, *, inplace: Literal[True]) -> None: ... @overload - def drop_duplicates(self, keep=..., inplace: bool = ...) -> Series | None: + def drop_duplicates( + self, keep: str | bool = ..., inplace: bool = ... + ) -> Series | None: ... - def drop_duplicates(self, keep="first", inplace=False) -> Series | None: + def drop_duplicates( + self, keep: str | bool = "first", inplace=False + ) -> Series | None: """ Return Series with duplicate values removed.