File tree Expand file tree Collapse file tree 2 files changed +3
-4
lines changed Expand file tree Collapse file tree 2 files changed +3
-4
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change @@ -188,7 +188,7 @@ def gpu_datetime_table(datetime_table):
188188
189189@pytest .fixture ()
190190def 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 ()
You can’t perform that action at this time.
0 commit comments