Failing test on CI test_dask_kneighbors_classifier::test_predict_proba

[jakirkham]

Seeing this failure on CI:

FAILED test_dask_kneighbors_classifier.py::test_predict_proba[dataset0-parameters0-dask_array] - assert 0.6 > 0.95
 +  where 0.6 = <function mean at 0x7f88a5f89af0>(array([[1., 0...dtype=float32) == array([[0., 0...dtype=float32)
 +    where <function mean at 0x7f88a5f89af0> = np.mean
    Full diff:
    - array([[0., 0., 1., 0., 0.],
    ?                 ----
    + array([[1., 0., 0., 0., 0.],
    ?         ++++
    +        [1., 0., 0., 0., 0.],
    -        [0., 0., 0., 0., 1.],
    ?                     ^   ^
    +        [0., 0., 0., 1., 0.],
    ?                     ^   ^
    -        [0., 1., 0., 0., 0.],
             ...,
    -        [1., 0., 0., 0., 0.],
    -        [0., 0., 0., 0., 1.],
    ?                     ^   ^
    +        [0., 0., 0., 1., 0.],
    ?                     ^   ^
    +        [0., 0., 1., 0., 0.],
    -        [0., 0., 0., 0., 1.]], dtype=float32,
    ?                      ----
    +        [0., 0., 1., 0., 0.]], dtype=float32,
    ?                 ++++
      ))

[jakirkham]

Fixed by rapidsai/cudf#15340