fixed failing unit test

pyronear · Dec 8, 2020 · 48dc220 · 48dc220
1 parent 13a0882
commit 48dc220
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_models.py b/test/test_models.py
@@ -71,9 +71,9 @@ def test_train_random_forest(self):
     def test_xgb_model(self):
         data = load_breast_cancer()
         a, b, c, d = score_v0.split_train_test(data['data'], data['target'])
-        _, _, preds = score_v0.xgb_model(a, c, b, d)
+        _, _, preds = score_v0.xgb_model(a, c, b, d, params={'min_child_weight': 5, 'eta': .1})
         self.assertEqual(len(preds), 114)
-        self.assertEqual(preds[0].round(3), np.float32(0.865))
+        self.assertEqual(preds[0].round(3), np.float32(0.996))
 
 
 if __name__ == "__main__":