From db3915c25c9ca5404df52dc56c1be1df658882f6 Mon Sep 17 00:00:00 2001 From: James Lamb Date: Sun, 27 Jun 2021 13:54:14 +0100 Subject: [PATCH] [tests][dask] add missing compute() in Dask test (#4412) --- tests/python_package_test/test_dask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python_package_test/test_dask.py b/tests/python_package_test/test_dask.py index 34290f1e60ef..4a3313b90da2 100644 --- a/tests/python_package_test/test_dask.py +++ b/tests/python_package_test/test_dask.py @@ -280,7 +280,7 @@ def test_classifier(output, task, boosting_type, tree_learner, cluster): pred_early_stop_margin=1.0, pred_early_stop_freq=2, raw_score=True - ) + ).compute() p1_proba = dask_classifier.predict_proba(dX).compute() p1_pred_leaf = dask_classifier.predict(dX, pred_leaf=True) p1_local = dask_classifier.to_local().predict(X)