Merge pull request #1276 from rapidsai/JohnZed-patch-1

Fix recent prediction crash in test_random_forest
rapidsai · Oct 16, 2019 · 0b320f0 · 0b320f0
2 parents d2c0e65 + a5832f1
commit 0b320f0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -64,6 +64,7 @@
 - PR #1236: Improve fix for SMO solvers potential crash on Turing
 - PR #1251: Disable compiler optimization for CUDA 10.1 for distance prims
 - PR #1260: Small bugfix for major conversion in input_utils
+- PR #1276: Fix float64 prediction crash in test_random_forest
 
 # cuML 0.9.0 (21 Aug 2019)
 

diff --git a/python/cuml/test/test_random_forest.py b/python/cuml/test/test_random_forest.py
@@ -47,7 +47,7 @@ def stress_param(*args, **kwargs):
                          stress_param(100)])
 @pytest.mark.parametrize('rows_sample', [unit_param(0.8), quality_param(0.85),
                          stress_param(0.9)])
-@pytest.mark.parametrize('datatype', [np.float32, np.float64])
+@pytest.mark.parametrize('datatype', [np.float32])
 @pytest.mark.parametrize('split_algo', [0, 1])
 @pytest.mark.parametrize('max_features', [1.0, 'auto', 'log2', 'sqrt'])
 def test_rf_classification(datatype, split_algo,
@@ -102,7 +102,7 @@ def test_rf_classification(datatype, split_algo,
                          stress_param(100)])
 @pytest.mark.parametrize('rows_sample', [unit_param(0.8), quality_param(0.85),
                          stress_param(0.9)])
-@pytest.mark.parametrize('datatype', [np.float32, np.float64])
+@pytest.mark.parametrize('datatype', [np.float32])
 @pytest.mark.parametrize('split_algo', [0, 1])
 @pytest.mark.parametrize('max_features', [1.0, 'auto', 'log2', 'sqrt'])
 def test_rf_regression(datatype, split_algo, rows_sample,