fix flakey test

kashif · Sep 13, 2024 · 2c3ac40 · 2c3ac40
1 parent 8a18417
commit 2c3ac40
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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