Avoid external API

rapidsai · Dec 11, 2023 · 2258a73 · 2258a73
1 parent 125e0d4
commit 2258a73
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/python/cudf/cudf/core/dataframe.py b/python/cudf/cudf/core/dataframe.py
@@ -2748,7 +2748,9 @@ def reindex(
             df = self
         else:
             columns = as_index(columns)
-            intersection = self.columns.intersection(columns.to_pandas())
+            intersection = self._data.to_pandas_index().intersection(
+                columns.to_pandas()
+            )
             df = self.loc[:, intersection]
 
         return df._reindex(

diff --git a/python/cudf/cudf/core/indexed_frame.py b/python/cudf/cudf/core/indexed_frame.py
@@ -2663,6 +2663,8 @@ def _reindex(
             else:
                 multiindex = False
                 names = column_names
+                if isinstance(names, cudf.Index):
+                    names = names.to_pandas()
                 rangeindex = isinstance(
                     column_names, (pd.RangeIndex, cudf.RangeIndex)
                 )