Skip to content

Commit 4604bee

Browse files
TYP: update EA.view() type annotation to indicate it returns an EA without dtype parameter
1 parent 7f670c1 commit 4604bee

File tree

3 files changed

+14
-2
lines changed

3 files changed

+14
-2
lines changed

pandas/core/arrays/_mixins.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,12 @@ def _validate_scalar(self, value):
113113

114114
# ------------------------------------------------------------------------
115115

116+
@overload
117+
def view(self) -> Self: ...
118+
119+
@overload
120+
def view(self, dtype: Dtype | None = ...) -> ArrayLike: ...
121+
116122
def view(self, dtype: Dtype | None = None) -> ArrayLike:
117123
# We handle datetime64, datetime64tz, timedelta64, and period
118124
# dtypes here. Everything else we pass through to the underlying

pandas/core/arrays/base.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1804,6 +1804,12 @@ def copy(self) -> Self:
18041804
"""
18051805
raise AbstractMethodError(self)
18061806

1807+
@overload
1808+
def view(self) -> ExtensionArray: ...
1809+
1810+
@overload
1811+
def view(self, dtype: Dtype | None = ...) -> ArrayLike: ...
1812+
18071813
def view(self, dtype: Dtype | None = None) -> ArrayLike:
18081814
"""
18091815
Return a view on the array.

pandas/core/arrays/string_.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -546,10 +546,10 @@ def _str_map_nan_semantics(
546546
else:
547547
return self._str_map_str_or_object(dtype, na_value, arr, f, mask)
548548

549-
def view(self, dtype: Dtype | None = None) -> ArrayLike:
549+
def view(self, dtype: Dtype | None = None) -> Self:
550550
if dtype is not None:
551551
raise TypeError("Cannot change data-type for string array.")
552-
return super().view(dtype=dtype)
552+
return super().view()
553553

554554

555555
# error: Definition of "_concat_same_type" in base class "NDArrayBacked" is

0 commit comments

Comments
 (0)