From 604ae01b7a3ed4e90e877b81859be0f137768a12 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Mon, 5 Apr 2021 18:32:04 +0800 Subject: [PATCH] [back port] Use CPU input for test_boost_from_prediction. (#6818) (#6824) --- tests/python-gpu/test_gpu_with_dask.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/python-gpu/test_gpu_with_dask.py b/tests/python-gpu/test_gpu_with_dask.py index cd48ccb4d..9295cb46a 100644 --- a/tests/python-gpu/test_gpu_with_dask.py +++ b/tests/python-gpu/test_gpu_with_dask.py @@ -174,12 +174,11 @@ def run_gpu_hist( def test_boost_from_prediction(local_cuda_cluster: LocalCUDACluster) -> None: - import cudf from sklearn.datasets import load_breast_cancer with Client(local_cuda_cluster) as client: X_, y_ = load_breast_cancer(return_X_y=True) - X = dd.from_array(X_, chunksize=100).map_partitions(cudf.from_pandas) - y = dd.from_array(y_, chunksize=100).map_partitions(cudf.from_pandas) + X = dd.from_array(X_, chunksize=100) + y = dd.from_array(y_, chunksize=100) run_boost_from_prediction(X, y, "gpu_hist", client)