From ad2f735a97f7d4e0f3c464731afa3c0521ffdf06 Mon Sep 17 00:00:00 2001 From: Lukas Drews Date: Fri, 12 Apr 2024 11:31:09 +0200 Subject: [PATCH] correct test --- tests/test_calibration.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_calibration.py b/tests/test_calibration.py index 3d9844ff..6f4440f2 100644 --- 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, )