diff --git a/requirements_test.txt b/requirements_test.txt index 1cf0dad255..a4aa11e714 100644 --- a/requirements_test.txt +++ b/requirements_test.txt @@ -41,7 +41,7 @@ timm==0.9.2 catboost==1.1.1 GPy==1.10.0 lightgbm==4.1.0 -xgboost==1.7.5 +xgboost==2.0.2 kornia~=0.6.12 tensorboardX==2.6.2.2 diff --git a/tests/metrics/test_verification_decision_trees.py b/tests/metrics/test_verification_decision_trees.py index 4f8588a0cd..fbbd4c2737 100644 --- a/tests/metrics/test_verification_decision_trees.py +++ b/tests/metrics/test_verification_decision_trees.py @@ -71,8 +71,8 @@ def test_XGBoost(self): x=self.x_test, y=self.y_test, eps_init=0.3, nb_search_steps=10, max_clique=2, max_level=2 ) - self.assertEqual(average_bound, 0.03186914062500001) - self.assertEqual(verified_error, 0.99) + self.assertAlmostEqual(average_bound, second=0.03335742187499999, places=6) + self.assertEqual(verified_error, second=0.99) def test_LightGBM(self): train_data = lightgbm.Dataset(self.x_train, label=np.argmax(self.y_train, axis=1))