Skip to content

Commit

Permalink
Skip XGBoost autologging test that no longer applies to XGB > 1.4.2 (m…
Browse files Browse the repository at this point in the history
…lflow#4516)

* Skip test

Signed-off-by: dbczumar <corey.zumar@databricks.com>

* Remove duplicate import

Signed-off-by: dbczumar <corey.zumar@databricks.com>
  • Loading branch information
dbczumar authored Jul 1, 2021
1 parent 5841e4f commit cd20dec
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions tests/xgboost/test_xgboost_autolog.py
Original file line number Diff line number Diff line change
Expand Up @@ -315,6 +315,13 @@ def test_xgb_autolog_loads_model_from_artifact(bst_params, dtrain):


@pytest.mark.large
@pytest.mark.skipif(
Version(xgb.__version__) > Version("1.4.2"),
reason=(
"In XGBoost <= 1.4.2, linear boosters do not support `get_score()` for importance value"
" creation. In XGBoost > 1.4.2, all boosters support `get_score()`."
),
)
def test_xgb_autolog_does_not_throw_if_importance_values_not_supported(dtrain):
# the gblinear booster does not support calling get_score on it,
# where get_score is used to create the importance values plot.
Expand Down

0 comments on commit cd20dec

Please sign in to comment.