Skip to content

Commit e42f3b5

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

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tmva/tmva/test/rbdt_xgboost.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,8 @@ def test_XGBMulticlass_default(self):
8282
"""
8383
Test model trained with multiclass XGBClassifier.
8484
"""
85+
if xgboost.__version__ >= "3.1.0":
86+
self.skipTest("We don't support multiclassification with xgboost>=3.1.0 yet")
8587
_test_XGBMulticlass("default")
8688

8789
def test_XGBRegression_default(self):

0 commit comments

Comments
 (0)