Merge pull request #2362 from Trusted-AI/dependabot/pip/xgboost-2.0.2

beat-buesser · web-flow · commit d7e91dd2b844 · 2023-12-20T12:38:54.000+01:00
Bump xgboost from 1.7.5 to 2.0.2
diff --git 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
@@ -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))

Original file line number	Diff line number	Diff line change
`@@ -71,8 +71,8 @@ def test_XGBoost(self):`
`71`	`71`	`x=self.x_test, y=self.y_test, eps_init=0.3, nb_search_steps=10, max_clique=2, max_level=2`
`72`	`72`	`)`
`73`	`73`
`74`		`- self.assertEqual(average_bound, 0.03186914062500001)`
`75`		`- self.assertEqual(verified_error, 0.99)`
	`74`	`+ self.assertAlmostEqual(average_bound, second=0.03335742187499999, places=6)`
	`75`	`+ self.assertEqual(verified_error, second=0.99)`
`76`	`76`
`77`	`77`	`def test_LightGBM(self):`
`78`	`78`	`train_data = lightgbm.Dataset(self.x_train, label=np.argmax(self.y_train, axis=1))`