Skip to content

Commit 2d92c34

Browse files
committed
mypy fix
1 parent c86f78d commit 2d92c34

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

pandas/core/series.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3680,7 +3680,7 @@ def argsort(self, axis=0, kind="quicksort", order=None) -> Series:
36803680
null -1
36813681
Name: xy, dtype: int64
36823682
"""
3683-
values = self.array
3683+
values = self.values
36843684
na_mask = isna(values)
36853685
if not any(na_mask):
36863686
res = np.argsort(values, kind=kind)
@@ -3707,7 +3707,11 @@ def argsort(self, axis=0, kind="quicksort", order=None) -> Series:
37073707
)
37083708
from pandas.core.reshape.concat import concat
37093709

3710-
return concat([notna_res_ser, na_res_ser])
3710+
ret_ser = concat([notna_res_ser, na_res_ser]).__finalize__(
3711+
self, method="argsort"
3712+
)
3713+
assert isinstance(ret_ser, Series) # mypy: concat 2 Series so is OK
3714+
return ret_ser
37113715

37123716
def nlargest(self, n=5, keep="first") -> Series:
37133717
"""

0 commit comments

Comments
 (0)