fixes for delete, isin, unique

pandas-dev · Dr-Irv · Jan 23, 2021 · Jan 23, 2021 · Jan 31, 2021 · Jan 31, 2021
commit 03b2c4a09acb3484cfc55b5253f255d4ed26e0c1
diff --git a/pandas/core/arrays/base.py b/pandas/core/arrays/base.py
@@ -776,7 +776,7 @@ def shift(self, periods: int = 1, fill_value: object = None) -> ExtensionArray:
             b = empty
         return self._concat_same_type([a, b])
 
-    def unique(self) -> "ExtensionArray":
+    def unique(self) -> ExtensionArray:
         """
         Compute the ExtensionArray of unique values.
 
@@ -873,7 +873,7 @@ def equals(self, other: object) -> bool:
             equal_na = self.isna() & other.isna()
             return bool((equal_values | equal_na).all())
 
-    def isin(self, values) -> np.ndarray:
+    def isin(self, values: Union[ExtensionArray, Sequence[Any]]) -> np.ndarray:
         """
         Pointwise comparison for set containment in the given values.
 
@@ -1288,7 +1288,9 @@ def __hash__(self) -> int:
     # ------------------------------------------------------------------------
     # Non-Optimized Default Methods
 
-    def delete(self: ExtensionArrayT, loc) -> ExtensionArrayT:
+    def delete(
+        self: ExtensionArrayT, loc: Union[int, Sequence[int]]
+    ) -> ExtensionArrayT:
         indexer = np.delete(np.arange(len(self)), loc)
         return self.take(indexer)