typing failed, less typing

Author: topper-123

pandas/core/arrays/masked.py

@@ -93,8 +93,6 @@
 
 from pandas.compat.numpy import function as nv
 
-_BaseMaskedArrayT = TypeVar("_BaseMaskedArrayT", bound="BaseMaskedArray")
-
 
 class BaseMaskedArray(OpsMixin, ExtensionArray):
     """
@@ -1429,9 +1427,7 @@ def _groupby_op(
         return self._maybe_mask_result(res_values, result_mask)
 
 
-def transpose_homogenous_masked_arrays(
-    masked_arrays: list[_BaseMaskedArrayT],
-) -> list[_BaseMaskedArrayT]:
+def transpose_homogenous_masked_arrays(masked_arrays):
     """Transpose masked arrays in a list, but faster.
 
     Input should be a list of 1-dim masked arrays of equal length and all have the

pandas/core/frame.py

@@ -3596,8 +3596,10 @@ def transpose(self, *args, copy: bool = False) -> DataFrame:
             if isinstance(dtypes[0], BaseMaskedDtype):
                 # We have masked arrays with the same dtype. We can transpose faster.
                 from pandas.core.arrays.masked import transpose_homogenous_masked_arrays
-
-                masked_arrays = [blk.values for blk in self._mgr.blocks]
+                if isinstance(self._mgr, ArrayManager):
+                    masked_arrays = [arr for arr in self._mgr.arrays]
+                else:
+                    masked_arrays = [blk.values for blk in self._mgr.blocks]
                 new_values = transpose_homogenous_masked_arrays(masked_arrays)
             else:
                 # We have other EAs with the same dtype. We preserve dtype in transpose.