wip

Author: MarcoGorelli

pandas/core/interchange/column.py

@@ -205,7 +205,9 @@ def describe_null(self):
                 try:
                     null, value = _NULL_DESCRIPTION[kind]
                 except KeyError as err:
-                    raise NotImplementedError(f"Data type {kind} not yet supported") from err
+                    raise NotImplementedError(
+                        f"Data type {kind} not yet supported"
+                    ) from err
 
                 return null, value
             return column_null_dtype, null_value
@@ -296,16 +298,6 @@ def _get_data_buffer(
         """
         Return the buffer containing the data and the buffer's associated dtype.
         """
-        if isinstance(self._col.dtype, ArrowDtype):
-            arr = self._col.array
-            buffer = PandasPyarrowBackedBuffer(arr._pa_array.chunks[0].buffers()[1])
-            dtype = (
-                DtypeKind.BOOL,
-                8,
-                ArrowCTypes.BOOL,
-                Endianness.NATIVE,
-            )  # note: currently only support native endianness
-            return buffer, dtype
         if self.dtype[0] == DtypeKind.DATETIME:
             # self.dtype[2] is an ArrowCTypes.TIMESTAMP where the tz will make
             # it longer than 4 characters
@@ -326,13 +318,17 @@ def _get_data_buffer(
             DtypeKind.FLOAT,
             DtypeKind.BOOL,
         ):
+            dtype = self.dtype
             arr = self._col.array
+            if isinstance(self._col.dtype, ArrowDtype):
+                arr = self._col.array
+                buffer = PandasPyarrowBackedBuffer(arr._pa_array.chunks[0].buffers()[1])
+                return buffer, dtype
             if isinstance(self._col.dtype, BaseMaskedDtype):
                 np_arr = arr._data  # type: ignore[attr-defined]
             else:
                 np_arr = arr._ndarray  # type: ignore[attr-defined]
             buffer = PandasBuffer(np_arr, allow_copy=self._allow_copy)
-            dtype = self.dtype
         elif self.dtype[0] == DtypeKind.CATEGORICAL:
             codes = self._col.values._codes
             buffer = PandasBuffer(codes, allow_copy=self._allow_copy)