solved mypy error in base.py (#1285)

pymc-labs · Dec 16, 2024 · 3aeb56e · 3aeb56e
1 parent 95db52b
commit 3aeb56e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc_marketing/mmm/base.py b/pymc_marketing/mmm/base.py
@@ -549,7 +549,7 @@ def _plot_group_predictive(
             hdi_list = [0.94, 0.5]
 
         if hdi_list and not add_gradient:
-            alpha_list = np.linspace(0.2, 0.4, len(hdi_list))
+            alpha_list = np.linspace(0.2, 0.4, len(hdi_list), dtype=float)
             for hdi_prob, alpha in zip(hdi_list, alpha_list, strict=True):
                 ax = self._add_hdi_to_plot(
                     ax=ax,