diff --git a/python/cuml/tests/test_random_forest.py b/python/cuml/tests/test_random_forest.py index 591dc515ac..1a77b14d05 100644 --- a/python/cuml/tests/test_random_forest.py +++ b/python/cuml/tests/test_random_forest.py @@ -1156,6 +1156,10 @@ def predict_with_json_rf_regressor(rf, x): np.testing.assert_almost_equal(pred, expected_pred, decimal=6) +@pytest.mark.xfail( + reason="Needs refactoring/debugging due to sporadic failures" + "https://github.com/rapidsai/cuml/issues/5528" +) @pytest.mark.memleak @pytest.mark.parametrize("estimator_type", ["classification"]) def test_rf_host_memory_leak(large_clf, estimator_type): @@ -1196,6 +1200,10 @@ def test_rf_host_memory_leak(large_clf, estimator_type): assert (final_mem - initial_baseline_mem) < 2.4e6 +@pytest.mark.xfail( + reason="Needs refactoring/debugging due to sporadic failures" + "https://github.com/rapidsai/cuml/issues/5528" +) @pytest.mark.memleak @pytest.mark.parametrize("estimator_type", ["regression", "classification"]) @pytest.mark.parametrize("i", list(range(100)))