Skip to content

Commit

Permalink
Merge pull request #5653 from PrimozGodec/fix-xgboost
Browse files Browse the repository at this point in the history
xgboost: extend workaround till next xgboost release
  • Loading branch information
markotoplak authored Oct 18, 2021
2 parents 0013e5a + 3ed1fe7 commit 8ed85c8
Showing 1 changed file with 7 additions and 5 deletions.
12 changes: 7 additions & 5 deletions Orange/modelling/tests/test_xgb.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
import unittest
from typing import Callable, Union
from datetime import datetime

import xgboost

from Orange.data import Table
from Orange.evaluation import CrossValidation
Expand All @@ -28,9 +27,12 @@ def setUpClass(cls):
cls.housing = Table("housing")

def test_xgboost_workaround(self):
self.assertLess(datetime.today(), datetime(2021, 10, 15),
"\nCheck the new release of xgboost and revert this "
"commit if possible, or do something else if not.")
self.assertLessEqual(
list(map(int, xgboost.__version__.split("."))),
[1, 5, 0],
"\nCheck the new release of xgboost and revert this "
"commit if possible, or do something else if not."
)

@test_learners
def test_cls(self, learner_class: Union[XGBLearner, XGBRFLearner]):
Expand Down

0 comments on commit 8ed85c8

Please sign in to comment.