Skip to content

Commit

Permalink
Revert "Use CPU input for test_boost_from_prediction. (dmlc#6818)"
Browse files Browse the repository at this point in the history
This reverts commit 74f3a2f.
  • Loading branch information
trivialfis committed Apr 14, 2021
1 parent b9a4f33 commit db462ec
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tests/python-gpu/test_gpu_with_dask.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)


Expand Down

0 comments on commit db462ec

Please sign in to comment.