Fix more typing

Author: mroeschke

pandas/core/indexes/multi.py

@@ -462,7 +462,9 @@ def _verify_integrity(
         result_codes = []
         for i in range(len(levels)):
             if i in levels_to_verify:
-                result_codes.append(self._validate_codes(levels[i], codes[i]))
+                # error: Argument 1 to "_validate_codes" of "MultiIndex"
+                # has incompatible type "Union[Any, Index]"; expected "List[Any]"
+                result_codes.append(self._validate_codes(levels[i], codes[i]))  # type: ignore[arg-type]  # noqa: E501
             else:
                 result_codes.append(codes[i])
 
@@ -772,9 +774,9 @@ def _values(self) -> np.ndarray:
             ):
                 vals = vals.astype(object)
 
-            vals = np.array(vals, copy=False)
-            vals = algos.take_nd(vals, codes, fill_value=index._na_value)
-            values.append(vals)
+            array_vals = np.array(vals, copy=False)
+            array_vals = algos.take_nd(array_vals, codes, fill_value=index._na_value)
+            values.append(array_vals)
 
         arr = lib.fast_zip(values)
         return arr
@@ -840,7 +842,7 @@ def size(self) -> int:
     # Levels Methods
 
     @cache_readonly
-    def levels(self) -> tuple:
+    def levels(self) -> tuple[Index, ...]:
         # Use cache_readonly to ensure that self.get_locs doesn't repeatedly
         # create new IndexEngine
         # https://github.com/pandas-dev/pandas/issues/31648

pandas/core/reshape/merge.py

@@ -1796,7 +1796,11 @@ def _convert_to_multiindex(index: Index) -> MultiIndex:
         join_codes = join_codes + [restore_codes]
         join_names = join_names + [dropped_level_name]
 
-    return join_levels, join_codes, join_names
+    # error: Incompatible return value type
+    # (got "Tuple[List[Index], List[Any], List[Any]]",
+    # expected "Tuple[List[Index], ndarray[Any, dtype[signedinteger[Any]]],
+    # List[Hashable]]")
+    return join_levels, join_codes, join_names  # type: ignore[return-value]
 
 
 class _OrderedMerge(_MergeOperation):
@@ -2222,8 +2226,10 @@ def _get_multiindex_indexer(
     join_keys: list[ArrayLike], index: MultiIndex, sort: bool
 ) -> tuple[npt.NDArray[np.intp], npt.NDArray[np.intp]]:
     # left & right join labels and num. of levels at each location
+    # error: Argument 1 to "_factorize_keys" has incompatible type "Index";
+    # expected "Union[ExtensionArray, ndarray[Any, Any]]"
     mapped = (
-        _factorize_keys(index.levels[n]._values, join_keys[n], sort=sort)
+        _factorize_keys(index.levels[n]._values, join_keys[n], sort=sort)  # type: ignore[arg-type]  # noqa: E501
         for n in range(index.nlevels)
     )
     zipped = zip(*mapped)

pandas/core/window/rolling.py

@@ -837,13 +837,17 @@ def _apply_pairwise(
             result_names = list(result.index.names)
         else:
             idx_codes, idx_levels = factorize(result.index)
+            idx_levels = cast(Index, idx_levels)
             result_codes = [idx_codes]
             result_levels = [idx_levels]
             result_names = [result.index.name]
 
         # 3) Create the resulting index by combining 1) + 2)
         result_codes = groupby_codes + result_codes
-        result_levels = groupby_levels + result_levels
+        # error: Incompatible types in assignment
+        # (expression has type "List[Union[Index, ndarray[Any, Any]]]",
+        # variable has type "List[Index]")
+        result_levels = groupby_levels + result_levels  # type: ignore[assignment]
         result_names = self._grouper.names + result_names
 
         result_index = MultiIndex(