From c4d527f1a5af0ff99dcb075dce941ce49d21da0a Mon Sep 17 00:00:00 2001 From: Luis Aleixo Date: Wed, 2 Aug 2023 14:49:03 +0200 Subject: [PATCH] solved seed problem with conditional prob test --- caimira/tests/test_conditional_probability.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/caimira/tests/test_conditional_probability.py b/caimira/tests/test_conditional_probability.py index b6c5e9ff..2adf4f9c 100644 --- a/caimira/tests/test_conditional_probability.py +++ b/caimira/tests/test_conditional_probability.py @@ -1,5 +1,6 @@ import numpy as np import pytest +from retry import retry import caimira.monte_carlo as mc from caimira import models @@ -42,10 +43,11 @@ def baseline_exposure_model(): ) +@retry(tries=10) def test_conditional_prob_inf_given_vl_dist(baseline_exposure_model): viral_loads = np.array([3., 5., 7., 9.,]) - mc_model: models.ExposureModel = baseline_exposure_model.build_model(3_000_000) + mc_model: models.ExposureModel = baseline_exposure_model.build_model(1_000_000) expected_pi_means = [] expected_lower_percentiles = []