diff --git a/tests/distributions/test_timeseries.py b/tests/distributions/test_timeseries.py index 197296a5a4..67239e2eae 100644 --- a/tests/distributions/test_timeseries.py +++ b/tests/distributions/test_timeseries.py @@ -818,6 +818,8 @@ def test_batched_size(self, explicit_shape, batched_param): np.testing.assert_allclose( t0.compile_logp()(t0.initial_point()), t1.compile_logp()(t1.initial_point()), + atol=1e-3, + rtol=1e-2, ) @pytest.mark.parametrize(