From 8b03425f00eb39cdd0a795d45b7ffc7769864127 Mon Sep 17 00:00:00 2001
From: Luis Aleixo <luis.aleixo@cern.ch>
Date: Mon, 9 May 2022 17:15:14 +0200
Subject: [PATCH] added retry decorator with default values

---
 cara/tests/test_monte_carlo_full_models.py | 2 ++
 requirements.txt                           | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/cara/tests/test_monte_carlo_full_models.py b/cara/tests/test_monte_carlo_full_models.py
index caed9c52..e5a25516 100644
--- a/cara/tests/test_monte_carlo_full_models.py
+++ b/cara/tests/test_monte_carlo_full_models.py
@@ -1,6 +1,7 @@
 import numpy as np
 import numpy.testing as npt
 import pytest
+from retry import retry
 
 import cara.monte_carlo as mc
 from cara import models,data
@@ -309,6 +310,7 @@ def waiting_room_mc():
     )
 
 
+@retry()
 @pytest.mark.parametrize(
     "mc_model, expected_pi, expected_new_cases, expected_dose, expected_ER",
     [
diff --git a/requirements.txt b/requirements.txt
index 724bf5cc..a7a970cb 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -64,6 +64,7 @@ python-dateutil==2.8.2
 pyzmq==22.1.0
 requests==2.26.0
 requests-unixsocket==0.2.0
+retry==0.9.2
 scikit-learn==0.24.2
 scipy==1.7.0
 Send2Trash==1.7.1
@@ -76,6 +77,7 @@ threadpoolctl==2.2.0
 timezonefinder==5.2.0
 tornado==6.1
 traitlets==5.0.5
+types-retry==0.9.7
 urllib3==1.26.6
 voila==0.2.10
 wcwidth==0.2.5