infinity anb beyond test was failing in torch==1.0.1 but not in torch…

…_nightly, acceptable tolerance differs
geoopt · Mar 3, 2019 · 53514cd · 53514cd
1 parent 673bfe1
commit 53514cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_poincare_math.py b/tests/test_poincare_math.py
@@ -343,7 +343,7 @@ def test_add_infinity_and_beyond(a, b, c):
     z = poincare.math.expmap(a, -infty, c=c)
     # they just need to be very far, exact answer is not supposed
     tolerance = {
-        torch.float32: dict(rtol=3e-1, atol=1e-1),
+        torch.float32: dict(rtol=3e-1, atol=2e-1),
         torch.float64: dict(rtol=1e-1, atol=1e-3),
     }
     np.testing.assert_allclose(z, -a, **tolerance[c.dtype])