diff --git a/tensorflow_probability/python/experimental/fastgp/schur_complement_test.py b/tensorflow_probability/python/experimental/fastgp/schur_complement_test.py index 3312894a01..713a88ead0 100644 --- a/tensorflow_probability/python/experimental/fastgp/schur_complement_test.py +++ b/tensorflow_probability/python/experimental/fastgp/schur_complement_test.py @@ -78,7 +78,7 @@ def testValuesAreCorrect(self, dims): np.testing.assert_allclose( expected_k.matrix(x, y2), actual_k.matrix(x, y2), - rtol=3e-3) + rtol=2e-2) class SchurComplementTestFloat32(_SchurComplementTest):