mypy, arraymanager fixups

Author: jbrockmendel

pandas/core/internals/array_manager.py

@@ -758,9 +758,9 @@ def fast_xs(self, loc: int) -> SingleArrayManager:
             result = dtype.construct_array_type()._from_sequence(values, dtype=dtype)
         # for datetime64/timedelta64, the np.ndarray constructor cannot handle pd.NaT
         elif is_datetime64_ns_dtype(dtype):
-            result = DatetimeArray._from_sequence(values, dtype=dtype)._data
+            result = DatetimeArray._from_sequence(values, dtype=dtype)._ndarray
         elif is_timedelta64_ns_dtype(dtype):
-            result = TimedeltaArray._from_sequence(values, dtype=dtype)._data
+            result = TimedeltaArray._from_sequence(values, dtype=dtype)._ndarray
         else:
             result = np.array(values, dtype=dtype)
         return SingleArrayManager([result], [self._axes[1]])

pandas/core/ops/__init__.py

@@ -6,7 +6,10 @@
 from __future__ import annotations
 
 import operator
-from typing import TYPE_CHECKING
+from typing import (
+    TYPE_CHECKING,
+    cast,
+)
 import warnings
 
 import numpy as np
@@ -423,6 +426,7 @@ def flex_arith_method_FRAME(op):
     @Appender(doc)
     def f(self, other, axis: Axis = "columns", level=None, fill_value=None):
         axis = self._get_axis_number(axis) if axis is not None else 1
+        axis = cast(int, axis)
 
         if should_reindex_frame_op(self, other, op, axis, fill_value, level):
             return frame_arith_method_with_reindex(self, other, op)