Merge pull request #3244 from rapidsai/branch-0.17

[gpuCI] Auto-merge branch-0.17 to branch-0.18 [skip ci]
rapidsai · Dec 3, 2020 · 50df6f6 · 50df6f6
2 parents ff17722 + 1287e86
commit 50df6f6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -105,6 +105,7 @@
 - PR #3208: Fix EXITCODE override in notebook test script
 - PR #3214: Correct flaky silhouette score test by setting atol
 - PR #3216: Ignore splits that do not satisfy constraints
+- PR #3239: Fix intermittent dask random forest failure
 
 # cuML 0.16.0 (23 Oct 2020)
 

diff --git a/python/cuml/test/dask/test_random_forest.py b/python/cuml/test/dask/test_random_forest.py
@@ -320,7 +320,7 @@ def test_rf_concatenation_dask(client, model_type):
 @pytest.mark.parametrize('model_type', ['classification', 'regression'])
 @pytest.mark.parametrize('ignore_empty_partitions', [True, False])
 def test_single_input(client, model_type, ignore_empty_partitions):
-    X, y = make_classification(n_samples=1)
+    X, y = make_classification(n_samples=1, n_classes=1)
     X = X.astype(np.float32)
     if model_type == 'classification':
         y = y.astype(np.int32)