From 65ea42bd422da411f29bcc7fa82c3306e32d092e Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Sun, 18 Oct 2020 19:16:46 -0700 Subject: [PATCH] [CI] Reduce testing load with RMM (#6249) * [CI] Reduce testing load with RMM * Address reviewer's comment --- tests/python-gpu/conftest.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/python-gpu/conftest.py b/tests/python-gpu/conftest.py index e493108a3..1e5e96df7 100644 --- a/tests/python-gpu/conftest.py +++ b/tests/python-gpu/conftest.py @@ -41,3 +41,15 @@ def local_cuda_cluster(request, pytestconfig): def pytest_addoption(parser): parser.addoption('--use-rmm-pool', action='store_true', default=False, help='Use RMM pool') + +def pytest_collection_modifyitems(config, items): + if config.getoption('--use-rmm-pool'): + blocklist = [ + 'python-gpu/test_gpu_demos.py::test_dask_training', + 'python-gpu/test_gpu_prediction.py::TestGPUPredict::test_shap', + 'python-gpu/test_gpu_linear.py::TestGPULinear' + ] + skip_mark = pytest.mark.skip(reason='This test is not run when --use-rmm-pool flag is active') + for item in items: + if any(item.nodeid.startswith(x) for x in blocklist): + item.add_marker(skip_mark)