scale bounds

pymc-labs · Aug 21, 2024 · 076dd2e · 076dd2e
1 parent a196a72
commit 076dd2e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pymc_marketing/mmm/budget_optimizer.py b/pymc_marketing/mmm/budget_optimizer.py
@@ -176,6 +176,11 @@ def allocate_budget(
                 "No budget bounds provided. Using default bounds (0, scaled_budget) for each channel.",
                 stacklevel=2,
             )
+        elif isinstance(budget_bounds, dict):
+            budget_bounds = {
+                channel: (bound[0] / self.scales.max(), bound[1] / self.scales.max())
+                for (channel, bound) in budget_bounds.items()
+            }
         elif not isinstance(budget_bounds, dict):
             raise TypeError("`budget_bounds` should be a dictionary.")