TYPING: Small fixes to make stubgen happy (#25576)

pandas-dev · Mar 8, 2019 · 0d893d0 · 0d893d0
1 parent 5c159d2
commit 0d893d0
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/pandas/core/arrays/period.py b/pandas/core/arrays/period.py
@@ -710,7 +710,7 @@ def _raise_on_incompatible(left, right):
 # Constructor Helpers
 
 def period_array(data, freq=None, copy=False):
-    # type: (Sequence[Optional[Period]], Optional[Tick]) -> PeriodArray
+    # type: (Sequence[Optional[Period]], Optional[Tick], bool) -> PeriodArray
     """
     Construct a new PeriodArray from a sequence of Period scalars.
 

diff --git a/pandas/core/arrays/sparse.py b/pandas/core/arrays/sparse.py
@@ -397,6 +397,7 @@ def _get_fill(arr):
 
 
 def _sparse_array_op(left, right, op, name):
+    # type: (SparseArray, SparseArray, Callable, str) -> Any
     """
     Perform a binary operation between two arrays.
 
@@ -413,7 +414,6 @@ def _sparse_array_op(left, right, op, name):
     -------
     SparseArray
     """
-    # type: (SparseArray, SparseArray, Callable, str) -> Any
     if name.startswith('__'):
         # For lookups in _libs.sparse we need non-dunder op name
         name = name[2:-2]

diff --git a/pandas/core/groupby/groupby.py b/pandas/core/groupby/groupby.py
@@ -1041,7 +1041,7 @@ def _bool_agg(self, val_test, skipna):
         """
 
         def objs_to_bool(vals):
-            # type: np.ndarray -> (np.ndarray, typing.Type)
+            # type: (np.ndarray) -> (np.ndarray, typing.Type)
             if is_object_dtype(vals):
                 vals = np.array([bool(x) for x in vals])
             else:
@@ -1743,7 +1743,7 @@ def quantile(self, q=0.5, interpolation='linear'):
         """
 
         def pre_processor(vals):
-            # type: np.ndarray -> (np.ndarray, Optional[typing.Type])
+            # type: (np.ndarray) -> (np.ndarray, Optional[typing.Type])
             if is_object_dtype(vals):
                 raise TypeError("'quantile' cannot be performed against "
                                 "'object' dtypes!")

diff --git a/pandas/core/internals/blocks.py b/pandas/core/internals/blocks.py
@@ -1827,13 +1827,13 @@ def interpolate(self, method='pad', axis=0, inplace=False, limit=None,
             placement=self.mgr_locs)
 
     def shift(self, periods, axis=0, fill_value=None):
+        # type: (int, Optional[BlockPlacement], Any) -> List[ExtensionBlock]
         """
         Shift the block by `periods`.
 
         Dispatches to underlying ExtensionArray and re-boxes in an
         ExtensionBlock.
         """
-        # type: (int, Optional[BlockPlacement]) -> List[ExtensionBlock]
         return [
             self.make_block_same_class(
                 self.values.shift(periods=periods, fill_value=fill_value),