Update test for XGBoost 2.x

Signed-off-by: Beat Buesser <[email protected]>
Trusted-AI · Dec 19, 2023 · 2badfc7 · 2badfc7
1 parent 60a94f2
commit 2badfc7
Showing 1 changed file with 2 additions and 2 deletions.
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))