correct test

scikit-learn-contrib · Apr 12, 2024 · ad2f735 · ad2f735
1 parent e4600d8
commit ad2f735
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_calibration.py b/tests/test_calibration.py
@@ -277,7 +277,10 @@ def test_beta_calibration(binary_calibration_data, binary_test_scores):
     assert proba.shape == (len(binary_test_scores),)
     assert_array_almost_equal(
         proba,
-        np.array([0.526125, 0.423743, 0.363907, 0.785855, 0.323201, 0.417089, 0.0]),
+        np.array(
+            [0.526125, 0.423743, 0.363907, 0.785855, 0.323201, 0.417089, 0.0],
+        ),
+        decimal=3,
     )