Skip to content

Commit ed25532

Browse files
authored
TYP: Add typing.overload signatures to DataFrame.drop_duplicates. (#54281)
* TYP: Add typing.overload signatures to DataFrame.drop_duplicates. This adds overloads so that a type checker can determine whether drop_duplicates returns DataFrame or None based on the value of the `inplace` argument. * Use 4-space indent.
1 parent 0e8c730 commit ed25532

File tree

1 file changed

+33
-0
lines changed

1 file changed

+33
-0
lines changed

pandas/core/frame.py

+33
Original file line numberDiff line numberDiff line change
@@ -6383,6 +6383,39 @@ def dropna(
63836383
self._update_inplace(result)
63846384
return None
63856385

6386+
@overload
6387+
def drop_duplicates(
6388+
self,
6389+
subset: Hashable | Sequence[Hashable] | None = ...,
6390+
*,
6391+
keep: DropKeep = ...,
6392+
inplace: Literal[True],
6393+
ignore_index: bool = ...,
6394+
) -> None:
6395+
...
6396+
6397+
@overload
6398+
def drop_duplicates(
6399+
self,
6400+
subset: Hashable | Sequence[Hashable] | None = ...,
6401+
*,
6402+
keep: DropKeep = ...,
6403+
inplace: Literal[False] = ...,
6404+
ignore_index: bool = ...,
6405+
) -> DataFrame:
6406+
...
6407+
6408+
@overload
6409+
def drop_duplicates(
6410+
self,
6411+
subset: Hashable | Sequence[Hashable] | None = ...,
6412+
*,
6413+
keep: DropKeep = ...,
6414+
inplace: bool = ...,
6415+
ignore_index: bool = ...,
6416+
) -> DataFrame | None:
6417+
...
6418+
63866419
def drop_duplicates(
63876420
self,
63886421
subset: Hashable | Sequence[Hashable] | None = None,

0 commit comments

Comments
 (0)