Skip to content

Commit 1e9eeba

Browse files
esmailansariesmail.ansariwilliambdeanpre-commit-ci[bot]
authored
Issue #1696: addresses a bug in plotting the intercept (#1698)
* Issue #1696: addresses a bug in plotting the intercept for time-varying-intercept * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: esmail.ansari <esmail.ansari@hrblock.com> Co-authored-by: Will Dean <57733339+williambdean@users.noreply.github.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 5ee48e6 commit 1e9eeba

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pymc_marketing/mmm/mmm.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1280,7 +1280,10 @@ def plot_components_contributions(
12801280

12811281
ax.plot(
12821282
np.asarray(self.X[self.date_column]),
1283-
np.full(len(self.X[self.date_column]), intercept.mean().data),
1283+
np.full(
1284+
len(self.X[self.date_column]),
1285+
intercept.mean(["chain", "draw"]).data,
1286+
),
12841287
color=f"C{i + 1}",
12851288
)
12861289
ax.fill_between(

0 commit comments

Comments
 (0)