Adjust tolerance levels in Sinkhorn solver tests for improved accuracy

Pierre-Louis Barbarant · Pierre-Louis Barbarant · commit 1694553b7078 · 2025-03-25T18:26:35.000+01:00
diff --git a/tests/solvers/test_utils.py b/tests/solvers/test_utils.py
@@ -76,7 +76,7 @@ def test_solvers_sinkhorn(pot_method, solver, is_log):
             alpha,
         )
         assert torch.allclose(log["beta"], beta)
-        assert torch.allclose(gamma, pi, atol=1e-5)
+        assert torch.allclose(gamma, pi)
 
 
 @pytest.mark.parametrize(
@@ -150,4 +150,4 @@ def test_solvers_sinkhorn_sparse(pot_method, solver, is_log):
             alpha,
         )
         assert torch.allclose(log["beta"], beta)
-        assert torch.allclose(gamma, pi.to_dense(), atol=1e-5)
+        assert torch.allclose(gamma, pi.to_dense(), atol=1e-6)

Original file line number	Diff line number	Diff line change
`@@ -76,7 +76,7 @@ def test_solvers_sinkhorn(pot_method, solver, is_log):`
`76`	`76`	`alpha,`
`77`	`77`	`)`
`78`	`78`	`assert torch.allclose(log["beta"], beta)`
`79`		`- assert torch.allclose(gamma, pi, atol=1e-5)`
	`79`	`+ assert torch.allclose(gamma, pi)`
`80`	`80`
`81`	`81`
`82`	`82`	`@pytest.mark.parametrize(`
`@@ -150,4 +150,4 @@ def test_solvers_sinkhorn_sparse(pot_method, solver, is_log):`
`150`	`150`	`alpha,`
`151`	`151`	`)`
`152`	`152`	`assert torch.allclose(log["beta"], beta)`
`153`		`- assert torch.allclose(gamma, pi.to_dense(), atol=1e-5)`
	`153`	`+ assert torch.allclose(gamma, pi.to_dense(), atol=1e-6)`