Annotations

Author: jbrockmendel

pandas/core/arraylike.py

@@ -36,7 +36,7 @@
 
 
 class OpsMixin:
-    def _supports_scalar_op(self, other, op_name: str):
+    def _supports_scalar_op(self, other, op_name: str) -> bool:
         """
         Return False to have unpack_zerodim_and_defer raise a TypeError with
         standardized exception message.
@@ -53,7 +53,7 @@ def _supports_scalar_op(self, other, op_name: str):
         """
         return True
 
-    def _supports_array_op(self, other: ArrayLike, op_name: str):
+    def _supports_array_op(self, other: ArrayLike, op_name: str) -> bool:
         """
         Return False to have unpack_zerodim_and_defer raise a TypeError with
         standardized exception message.

pandas/core/arrays/categorical.py

@@ -1612,10 +1612,10 @@ def map(
     __le__ = _cat_compare_op(operator.le)
     __ge__ = _cat_compare_op(operator.ge)
 
-    def _supports_scalar_op(self, other, op_name: str):
+    def _supports_scalar_op(self, other, op_name: str) -> bool:
         return True
 
-    def _supports_array_op(self, other, op_name: str):
+    def _supports_array_op(self, other: ArrayLike, op_name: str) -> bool:
         return True
 
     # -------------------------------------------------------------

pandas/core/arrays/interval.py

@@ -734,10 +734,10 @@ def __setitem__(self, key, value) -> None:
         self._left[key] = value_left
         self._right[key] = value_right
 
-    def _supports_scalar_op(self, other, op_name):
+    def _supports_scalar_op(self, other, op_name: str) -> bool:
         return True
 
-    def _supports_array_op(self, other, op_name):
+    def _supports_array_op(self, other: ArrayLike, op_name: str) -> bool:
         return True
 
     def _cmp_method(self, other, op):