Implement retry mechanism

helpers/thompson_sampling.py

@@ -6,6 +6,7 @@ import random
 import matplotlib.pyplot as plt
 import numpy as np
 import streamlit as st
+from tenacity import retry, stop_after_attempt, wait_fixed
 
 import config as cfg
 
@@ -61,6 +62,7 @@ class ThompsonSampler:
     def create_price_plot(self):
         fig = plt.figure()
         plt.xlabel("Price")
+        plt.xlim(0,20)
         plt.yticks(color='w')
 
         price_distr = [self.calc_optimal_price(post_demand, sample=False)
@@ -119,6 +121,7 @@ class ThompsonSampler:
         self.calc_optimal_price(posterior_sample, sample=True)
         self.create_plots(highlighted_sample=posterior_sample)
 
+    @retry(stop=stop_after_attempt(5), wait=wait_fixed(0.25))
     def run(self):
         if st.session_state.latent_elasticity != self.latent_elasticity:
             self.update_elasticity()

requirements.txt

@@ -1,3 +1,4 @@
 pymc3==3.11.5
 streamlit==1.10.0
-sympy==1.10.1
+sympy==1.10.1
+tenacity==8.1.0