Skip to content

Commit 1572229

Browse files
committed
[TVMA] Disable xgboost multiclassification test for xgboost>=3.1.0
This is not supported yet.
1 parent 199952f commit 1572229

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tmva/tmva/test/rbdt_xgboost.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import ROOT
88
import numpy as np
99
import json
10+
import pytest
1011

1112
np.random.seed(1234)
1213

@@ -78,6 +79,9 @@ def test_XGBBinary_default(self):
7879
"""
7980
_test_XGBBinary("default")
8081

82+
@pytest.mark.xfail(
83+
xgb.__version__ >= "3.1.0"
84+
reason="We don't support multiclassification with xgboost>=3.1.0 yet")
8185
def test_XGBMulticlass_default(self):
8286
"""
8387
Test model trained with multiclass XGBClassifier.

0 commit comments

Comments
 (0)