handle ColumnAccessor inputs in dataframe constructor

rapidsai · Feb 7, 2022 · ded9906 · ded9906
1 parent 2e458b9
commit ded9906
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/cudf/cudf/core/dataframe.py b/python/cudf/cudf/core/dataframe.py
@@ -568,6 +568,10 @@ def __init__(
                         for k in columns
                     }
                 )
+        elif isinstance(data, ColumnAccessor):
+            result = self._from_data(data, index, columns)
+            self._data = result._data
+            self._index = result._index
         elif hasattr(data, "__cuda_array_interface__"):
             arr_interface = data.__cuda_array_interface__