From db462ec9752fe0149efd27b7072158ce58b754b3 Mon Sep 17 00:00:00 2001 From: fis Date: Wed, 14 Apr 2021 17:19:29 +0800 Subject: [PATCH] Revert "Use CPU input for test_boost_from_prediction. (#6818)" This reverts commit 74f3a2f4b5c2654af90d1477fd543b5d97280fbe. --- tests/python-gpu/test_gpu_with_dask.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/python-gpu/test_gpu_with_dask.py b/tests/python-gpu/test_gpu_with_dask.py index cfdd9db12f98..2068ededc975 100644 --- a/tests/python-gpu/test_gpu_with_dask.py +++ b/tests/python-gpu/test_gpu_with_dask.py @@ -175,11 +175,12 @@ def run_gpu_hist( @pytest.mark.skipif(**tm.no_cudf()) 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) - y = dd.from_array(y_, chunksize=100) + X = dd.from_array(X_, chunksize=100).map_partitions(cudf.from_pandas) + y = dd.from_array(y_, chunksize=100).map_partitions(cudf.from_pandas) run_boost_from_prediction(X, y, "gpu_hist", client)