Call to_xarray on the slow object

Author: Matt711

python/cudf/cudf/pandas/_wrappers/pandas.py

@@ -315,6 +315,16 @@ def _DataFrame_columns(self):
     return result
 
 
+def _to_xarray(self):
+    # Call to_xarray directly on the slow object, not via _FastSlowAttribute.
+    # This keeps the module accelerator enabled so pandas returns proxy
+    # Index/ExtensionArray objects that inherit from ExtensionArray and pass
+    # xarray's isinstance checks. If those checks fail, xarray falls back to
+    # np.issubdtype(array.dtype, ...), which raises TypeError for unsupported
+    # extension dtypes.
+    return self._fsproxy_slow.to_xarray()
+
+
 DataFrame = make_final_proxy_type(
     "DataFrame",
     cudf.DataFrame,
@@ -347,6 +357,7 @@ def _DataFrame_columns(self):
         "flags": _FastSlowAttribute("flags", private=True),
         "memory_usage": _FastSlowAttribute("memory_usage"),
         "__sizeof__": _FastSlowAttribute("__sizeof__"),
+        "to_xarray": _to_xarray,
     },
 )
 
@@ -420,6 +431,7 @@ def _argsort(self, *args, **kwargs):
         "_accessors": set(),
         "dtype": property(_Series_dtype),
         "argsort": _argsort,
+        "to_xarray": _to_xarray,
         "attrs": _FastSlowAttribute("attrs"),
         "_mgr": _FastSlowAttribute("_mgr", private=True),
         "array": _FastSlowAttribute("array", private=True),
@@ -576,6 +588,7 @@ def Index__setattr__(self, name, value):
     pd.Categorical,
     fast_to_slow=_Unusable(),
     slow_to_fast=_Unusable(),
+    bases=(pd.api.extensions.ExtensionArray,),
 )
 
 CategoricalDtype = make_final_proxy_type(
@@ -615,6 +628,7 @@ def Index__setattr__(self, name, value):
     pd.arrays.DatetimeArray,
     fast_to_slow=_Unusable(),
     slow_to_fast=_Unusable(),
+    bases=(pd.api.extensions.ExtensionArray,),
     additional_attributes={
         "_data": _FastSlowAttribute("_data", private=True),
         "_mask": _FastSlowAttribute("_mask", private=True),
@@ -718,6 +732,7 @@ def Index__setattr__(self, name, value):
     pd.arrays.PeriodArray,
     fast_to_slow=_Unusable(),
     slow_to_fast=_Unusable(),
+    bases=(pd.api.extensions.ExtensionArray,),
     additional_attributes={
         "_data": _FastSlowAttribute("_data", private=True),
         "_mask": _FastSlowAttribute("_mask", private=True),
@@ -800,6 +815,7 @@ def Index__setattr__(self, name, value):
     pd.arrays.StringArray,
     fast_to_slow=_Unusable(),
     slow_to_fast=_Unusable(),
+    bases=(pd.api.extensions.ExtensionArray,),
     additional_attributes={
         "_data": _FastSlowAttribute("_data", private=True),
         "_mask": _FastSlowAttribute("_mask", private=True),
@@ -838,6 +854,7 @@ def Index__setattr__(self, name, value):
     pd.core.arrays.string_arrow.ArrowStringArray,
     fast_to_slow=_Unusable(),
     slow_to_fast=_Unusable(),
+    bases=(pd.api.extensions.ExtensionArray,),
     additional_attributes={
         "_pa_array": _FastSlowAttribute("_pa_array", private=True),
         "__array__": _FastSlowAttribute("__array__", private=True),
@@ -877,6 +894,7 @@ def Index__setattr__(self, name, value):
     pd.arrays.BooleanArray,
     fast_to_slow=_Unusable(),
     slow_to_fast=_Unusable(),
+    bases=(pd.api.extensions.ExtensionArray,),
     additional_attributes={
         "_data": _FastSlowAttribute("_data", private=True),
         "_mask": _FastSlowAttribute("_mask", private=True),
@@ -901,6 +919,7 @@ def Index__setattr__(self, name, value):
     pd.arrays.IntegerArray,
     fast_to_slow=_Unusable(),
     slow_to_fast=_Unusable(),
+    bases=(pd.api.extensions.ExtensionArray,),
     additional_attributes={
         "__array_ufunc__": _FastSlowAttribute("__array_ufunc__"),
         "_data": _FastSlowAttribute("_data", private=True),
@@ -1021,6 +1040,7 @@ def Index__setattr__(self, name, value):
     pd.arrays.IntervalArray,
     fast_to_slow=_Unusable(),
     slow_to_fast=_Unusable(),
+    bases=(pd.api.extensions.ExtensionArray,),
     additional_attributes={
         "_data": _FastSlowAttribute("_data", private=True),
         "_mask": _FastSlowAttribute("_mask", private=True),
@@ -1055,6 +1075,7 @@ def Index__setattr__(self, name, value):
     pd.arrays.FloatingArray,
     fast_to_slow=_Unusable(),
     slow_to_fast=_Unusable(),
+    bases=(pd.api.extensions.ExtensionArray,),
     additional_attributes={
         "__array_ufunc__": _FastSlowAttribute("__array_ufunc__"),
         "_data": _FastSlowAttribute("_data", private=True),

python/cudf/cudf/pandas/scripts/conftest-patch.py

@@ -3421,8 +3421,6 @@ def set_copy_on_write_option():
     "tests/generic/test_to_xarray.py::TestDataFrameToXArray::test_to_xarray_index_types[uint32]",
     "tests/generic/test_to_xarray.py::TestDataFrameToXArray::test_to_xarray_index_types[uint64]",
     "tests/generic/test_to_xarray.py::TestDataFrameToXArray::test_to_xarray_index_types[uint8]",
-    "tests/generic/test_to_xarray.py::TestSeriesToXArray::test_to_xarray_index_types[string-pyarrow]",
-    "tests/generic/test_to_xarray.py::TestSeriesToXArray::test_to_xarray_index_types[string-python]",
     "tests/groupby/aggregate/test_aggregate.py::test_agg_multiple_with_as_index_false_subset_to_a_single_column",
     "tests/groupby/aggregate/test_aggregate.py::test_agg_str_with_kwarg_axis_1_raises[count]",
     "tests/groupby/aggregate/test_aggregate.py::test_agg_str_with_kwarg_axis_1_raises[first]",