chore: fix test approx in metrics

amphibian-dev · Mar 24, 2024 · 4952f48 · 4952f48
1 parent f0eb704
commit 4952f48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/toad/metrics_test.py b/toad/metrics_test.py
@@ -56,11 +56,11 @@ def test_KS_bucket_with_cum_lift():
 
 def test_F1():
     result, split = F1(feature, target, return_split = True)
-    assert result == 0.6844207723035951
+    assert result == pytest.approx(0.6844207723035951, FUZZ_THRESHOLD)
 
 def test_F1_split():
     result = F1(feature, target, split = 0.5)
-    assert result == 0.51417004048583
+    assert result == pytest.approx(0.51417004048583, FUZZ_THRESHOLD)
 
 def test_AUC():
     result = AUC(feature, target)