[tests] Loosen precision requirements to GPU precision
@@ -697,5 +697,5 @@ class TestMathFunctions(TestCase):
y = math.stack([0.8 * x[0] - x[1] + 1, -0.8 * x[0]], channel(features='y0,y1'))
from phi.math._fit import fit_hyperplane
w, b = fit_hyperplane(x, y, 'batch')
- math.assert_close(w, math.wrap([(0.8, -1), (-0.8, 0)], channel(y), channel(x)), abs_tolerance=1e-5)
- math.assert_close(b, math.wrap((1, 0), channel(y)), abs_tolerance=1e-5)
+ math.assert_close(w, math.wrap([(0.8, -1), (-0.8, 0)], channel(y), channel(x)), abs_tolerance=1e-3)
+ math.assert_close(b, math.wrap((1, 0), channel(y)), abs_tolerance=1e-3)
