Adapt atol of test_equivalence_mean

ValerianRey · ValerianRey · commit e261e794852e · 2025-03-27T21:58:15.000+01:00
diff --git a/tests/unit/aggregation/test_cagrad.py b/tests/unit/aggregation/test_cagrad.py
@@ -33,7 +33,7 @@ def test_equivalence_mean(matrix: Tensor):
     result = ca_grad(matrix)
     expected = mean(matrix)
 
-    assert_close(result, expected, atol=1e-1, rtol=0)
+    assert_close(result, expected, atol=2e-1, rtol=0)
 
 
 @mark.parametrize(
