Skip to content

GitLab

Unverified Commit bf22a25d authored by Frank Fineis's avatar Frank Fineis Committed by GitHub
Browse files

[dask] Address flaky test_ranker tests (#3819)

parent b8fc476e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/python_package_test/test_dask.py
View file @ bf22a25d
...@@ -512,7 +512,7 @@ def test_ranker(output, client, listen_port, group): ...@@ -512,7 +512,7 @@ def test_ranker(output, client, listen_port, group):
# have high rank correlation with scores from serial ranker. # have high rank correlation with scores from serial ranker.
dcor = spearmanr(rnkvec_dask, y).correlation dcor = spearmanr(rnkvec_dask, y).correlation
assert dcor > 0.6 assert dcor > 0.6
assert spearmanr(rnkvec_dask, rnkvec_local).correlation > 0.9 assert spearmanr(rnkvec_dask, rnkvec_local).correlation > 0.75
client.close() client.close()
......
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