From 6d043253302d88d17fa92428574dcc86e02595a9 Mon Sep 17 00:00:00 2001 From: jburnim Date: Mon, 6 Jun 2022 11:09:37 -0700 Subject: [PATCH] Increase tolerance in LinearOperatorUnitaryTest.test_solve. PiperOrigin-RevId: 453239614 --- .../experimental/linalg/linear_operator_unitary_test.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tensorflow_probability/python/experimental/linalg/linear_operator_unitary_test.py b/tensorflow_probability/python/experimental/linalg/linear_operator_unitary_test.py index 361117519e..6c1ca76015 100644 --- a/tensorflow_probability/python/experimental/linalg/linear_operator_unitary_test.py +++ b/tensorflow_probability/python/experimental/linalg/linear_operator_unitary_test.py @@ -82,11 +82,13 @@ def test_solve(self): broadcasted_y = tf.broadcast_to(y, [7, 1, 13, 3]) self.assertAllClose( self.evaluate(tf.linalg.solve(q, broadcasted_y)), - self.evaluate(operator.solve(y))) + self.evaluate(operator.solve(y)), + atol=1e-5, rtol=1e-5) self.assertAllClose( self.evaluate(tf.linalg.solve(q, broadcasted_y, adjoint=True)), - self.evaluate(operator.solve(y, adjoint=True))) + self.evaluate(operator.solve(y, adjoint=True)), + atol=1e-5, rtol=1e-5) if __name__ == '__main__':