Unverified Commit db3915c2 authored by James Lamb's avatar James Lamb Committed by GitHub
Browse files

[tests][dask] add missing compute() in Dask test (#4412)

parent aacb4c8f
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment