Limit possible plot size

config.py

@@ -6,3 +6,5 @@ VARIABLE_COST = 0.2
 LATENT_ELASTICITY = 0.25
 LATENT_SHAPE = 2
 LATENT_STDEV = 0.25
+
+MAX_PRICE_SAMPLES = 20

helpers/thompson_sampling.py

@@ -109,6 +109,8 @@ class ThompsonSampler:
         optimal_price = self.possible_prices[np.argmax(profit)]
         if sample:
             self.price_samples.append(optimal_price)
+            if len(self.price_samples) > cfg.MAX_PRICE_SAMPLES:
+                self.price_samples.pop(0)
         return optimal_price
 
     def update_posteriors(self, samples=75):