Skip to content

Commit

Permalink
Skip rf_memleak tests (#5529)
Browse files Browse the repository at this point in the history
reference for tracking:  #5528

Authors:
  - Dante Gama Dessavre (https://github.com/dantegd)

Approvers:
  - William Hicks (https://github.com/wphicks)
  - Simon Adorf (https://github.com/csadorf)

URL: #5529
  • Loading branch information
dantegd authored Jul 28, 2023
1 parent e310957 commit 20781a5
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions python/cuml/tests/test_random_forest.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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)))
Expand Down

0 comments on commit 20781a5

Please sign in to comment.