Skip to content

Commit

Permalink
xgboost: extend workaround till next xgboost release
Browse files Browse the repository at this point in the history
  • Loading branch information
PrimozGodec committed Oct 18, 2021
1 parent 0013e5a commit 32b5199
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions Orange/modelling/tests/test_xgb.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
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 +29,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 32b5199

Please sign in to comment.