From 048acf81cd258ff1ab165fbb8313ce3652fa9107 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Sun, 1 Nov 2020 20:59:27 +0800 Subject: [PATCH] Enable shap sparse test. (#6332) --- tests/python-gpu/test_gpu_prediction.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tests/python-gpu/test_gpu_prediction.py b/tests/python-gpu/test_gpu_prediction.py index b22caf306..be7c1fdbe 100644 --- a/tests/python-gpu/test_gpu_prediction.py +++ b/tests/python-gpu/test_gpu_prediction.py @@ -212,10 +212,6 @@ class TestGPUPredict(unittest.TestCase): tm.dataset_strategy, shap_parameter_strategy) @settings(deadline=None, max_examples=20) def test_shap_interactions(self, num_rounds, dataset, param): - if dataset.name == 'sparse': - issue = 'https://github.com/dmlc/xgboost/issues/6074' - pytest.xfail(reason=f'GPU shap with sparse is flaky: {issue}') - param.update({"predictor": "gpu_predictor", "gpu_id": 0}) param = dataset.set_params(param) dmat = dataset.get_dmat()