From ded99065916e58a8454d910cbb9c740a26e99404 Mon Sep 17 00:00:00 2001 From: galipremsagar Date: Mon, 7 Feb 2022 10:05:04 -0800 Subject: [PATCH] handle ColumnAccessor inputs in dataframe constructor --- python/cudf/cudf/core/dataframe.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/cudf/cudf/core/dataframe.py b/python/cudf/cudf/core/dataframe.py index 25f2f71b40d..48406f74155 100644 --- 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__