Skip to content

Commit 7e02237

Browse files
Avoid using the to/from_dask_dataframe dask-cudf API (#1328)
* avoid using to/from_dask_dataframe * Linting --------- Co-authored-by: Charles Blackmon-Luca <[email protected]>
1 parent f1b4d93 commit 7e02237

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

dask_sql/input_utils/dask.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,11 +23,10 @@ def to_dc(
2323
):
2424
if gpu: # pragma: no cover
2525
try:
26-
import dask_cudf
26+
import dask_cudf # noqa: F401
2727
except ImportError:
2828
raise ModuleNotFoundError(
2929
"Setting `gpu=True` for table creation requires dask_cudf"
3030
)
31-
if not isinstance(input_item, dask_cudf.DataFrame):
32-
input_item = dask_cudf.from_dask_dataframe(input_item, **kwargs)
31+
return input_item.to_backend("cudf", **kwargs)
3332
return input_item

tests/integration/fixtures.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,7 @@ def gpu_datetime_table(datetime_table):
188188

189189
@pytest.fixture()
190190
def gpu_timeseries(timeseries):
191-
return dask_cudf.from_dask_dataframe(timeseries) if dask_cudf else None
191+
return timeseries.to_backend("cudf") if dask_cudf else None
192192

193193

194194
@pytest.fixture()

0 commit comments

Comments
 (0)