Skip to content

Commit 61951f2

Browse files
committed
duplicate overloads to make mypy pass
1 parent b4b69ef commit 61951f2

File tree

1 file changed

+34
-3
lines changed

1 file changed

+34
-3
lines changed

pandas-stubs/core/series.pyi

+34-3
Original file line numberDiff line numberDiff line change
@@ -309,10 +309,11 @@ class Series(IndexOpsMixin, NDFrame, Generic[S1]):
309309
def index(self) -> Index | MultiIndex: ...
310310
@index.setter
311311
def index(self, idx: Index) -> None: ...
312+
# TODO: combine Level | Sequence[Level] github.com/python/mypy/issues/14311
312313
@overload
313314
def reset_index(
314315
self,
315-
level: Sequence[Level] | Level | None = ...,
316+
level: Sequence[Level] = ...,
316317
*,
317318
drop: Literal[False] = ...,
318319
name: Level = ...,
@@ -322,7 +323,7 @@ class Series(IndexOpsMixin, NDFrame, Generic[S1]):
322323
@overload
323324
def reset_index(
324325
self,
325-
level: Sequence[Level] | Level | None = ...,
326+
level: Sequence[Level] = ...,
326327
*,
327328
drop: Literal[True],
328329
name: Level = ...,
@@ -332,7 +333,37 @@ class Series(IndexOpsMixin, NDFrame, Generic[S1]):
332333
@overload
333334
def reset_index(
334335
self,
335-
level: Sequence[Level] | Level | None = ...,
336+
level: Sequence[Level] = ...,
337+
*,
338+
drop: bool = ...,
339+
name: Level = ...,
340+
inplace: Literal[True],
341+
allow_duplicates: bool = ...,
342+
) -> None: ...
343+
@overload
344+
def reset_index(
345+
self,
346+
level: Level | None = ...,
347+
*,
348+
drop: Literal[False] = ...,
349+
name: Level = ...,
350+
inplace: Literal[False] = ...,
351+
allow_duplicates: bool = ...,
352+
) -> DataFrame: ...
353+
@overload
354+
def reset_index(
355+
self,
356+
level: Level | None = ...,
357+
*,
358+
drop: Literal[True],
359+
name: Level = ...,
360+
inplace: Literal[False] = ...,
361+
allow_duplicates: bool = ...,
362+
) -> Series: ...
363+
@overload
364+
def reset_index(
365+
self,
366+
level: Level | None = ...,
336367
*,
337368
drop: bool = ...,
338369
name: Level = ...,

0 commit comments

Comments
 (0)