lower atol

Lightning-AI · Oct 31, 2024 · 2bf58e4 · 2bf58e4
1 parent 6af149b
commit 2bf58e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unittests/classification/test_logauc.py b/tests/unittests/classification/test_logauc.py
@@ -159,7 +159,7 @@ def _multiclass_compare_implementation(preds, target, fpr_range, average):
 class TestMulticlassLogAUC(MetricTester):
     """Test class for `MulticlassLogAUC` metric."""
 
-    atol = 1e-4
+    atol = 1e-3
 
     @pytest.mark.parametrize("fpr_range", [(0.001, 0.1), (0.01, 0.1), (0.1, 0.2)])
     @pytest.mark.parametrize("average", ["macro", None])
@@ -278,7 +278,7 @@ def _multilabel_compare_implementation(preds, target, fpr_range, average):
 class TestMultilabelLogAUC(MetricTester):
     """Test class for `MultilabelLogAUC` metric."""
 
-    atol = 1e-4
+    atol = 1e-3
 
     @pytest.mark.parametrize("fpr_range", [(0.001, 0.1), (0.01, 0.1), (0.1, 0.2)])
     @pytest.mark.parametrize("average", ["macro", None])