Skip to content

GitLab

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
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/python_package_test/test_dask.py
View file @ db3915c2
...@@ -280,7 +280,7 @@ def test_classifier(output, task, boosting_type, tree_learner, cluster): ...@@ -280,7 +280,7 @@ def test_classifier(output, task, boosting_type, tree_learner, cluster):
pred_early_stop_margin=1.0, pred_early_stop_margin=1.0,
pred_early_stop_freq=2, pred_early_stop_freq=2,
raw_score=True raw_score=True
) ).compute()
p1_proba = dask_classifier.predict_proba(dX).compute() p1_proba = dask_classifier.predict_proba(dX).compute()
p1_pred_leaf = dask_classifier.predict(dX, pred_leaf=True) p1_pred_leaf = dask_classifier.predict(dX, pred_leaf=True)
p1_local = dask_classifier.to_local().predict(X) 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