From 2c3ac4002097630f9dee5f7f5252aa3bfc65c925 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Fri, 13 Sep 2024 20:08:08 +0200 Subject: [PATCH] fix flakey test --- test/mx/distribution/test_mx_distribution_inference.py | 1 + 1 file changed, 1 insertion(+) diff --git a/test/mx/distribution/test_mx_distribution_inference.py b/test/mx/distribution/test_mx_distribution_inference.py index b26a2d2ce7..753a084cf0 100644 --- a/test/mx/distribution/test_mx_distribution_inference.py +++ b/test/mx/distribution/test_mx_distribution_inference.py @@ -499,6 +499,7 @@ def test_dirichlet(hybridize: bool) -> None: ), f"Covariance did not match: cov = {cov}, cov_hat = {cov_hat}" +@pytest.mark.flaky(retries=3) @pytest.mark.parametrize("hybridize", [True, False]) def test_dirichlet_multinomial(hybridize: bool) -> None: num_samples = 2000