diff --git a/distrax/_src/bijectors/rational_quadratic_spline_test.py b/distrax/_src/bijectors/rational_quadratic_spline_test.py index cb10577..bad82d9 100644 --- a/distrax/_src/bijectors/rational_quadratic_spline_test.py +++ b/distrax/_src/bijectors/rational_quadratic_spline_test.py @@ -175,7 +175,7 @@ def test_composite_methods_are_consistent(self): logdet1 = self.variant(bijector.inverse_log_det_jacobian)(y) x2, logdet2 = self.variant(bijector.inverse_and_log_det)(y) np.testing.assert_allclose(x1, x2, atol=7e-8) - np.testing.assert_allclose(logdet1, logdet2, atol=7e-8) + np.testing.assert_allclose(logdet1, logdet2, atol=8e-7) @chex.all_variants def test_boundary_conditions(self):