Skip to content

Commit

Permalink
Enable shap sparse test. (#6332)
Browse files Browse the repository at this point in the history
  • Loading branch information
trivialfis committed Nov 1, 2020
1 parent 5e1e972 commit 048acf8
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions tests/python-gpu/test_gpu_prediction.py
Expand Up @@ -212,10 +212,6 @@ def test_shap(self, num_rounds, dataset, param):
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()
Expand Down

0 comments on commit 048acf8

Please sign in to comment.