File tree 2 files changed +3
-4
lines changed
2 files changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -23,11 +23,10 @@ def to_dc(
23
23
):
24
24
if gpu : # pragma: no cover
25
25
try :
26
- import dask_cudf
26
+ import dask_cudf # noqa: F401
27
27
except ImportError :
28
28
raise ModuleNotFoundError (
29
29
"Setting `gpu=True` for table creation requires dask_cudf"
30
30
)
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 )
33
32
return input_item
Original file line number Diff line number Diff line change @@ -188,7 +188,7 @@ def gpu_datetime_table(datetime_table):
188
188
189
189
@pytest .fixture ()
190
190
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
192
192
193
193
194
194
@pytest .fixture ()
You can’t perform that action at this time.
0 commit comments