Skip to content

Commit 717702a

Browse files
shuvayansangeeduttapre-commit-ci[bot]williambdean
authored
Future-proof prior_linearized method call (#806)
* modified: pymc_marketing/mmm/tvp.py * Update tvp.py Changed Xs to X in prior_linearized( as per the change in pymc) * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: sangeedutta <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Will Dean <[email protected]>
1 parent d235ff4 commit 717702a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc_marketing/mmm/tvp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,7 @@ def time_varying_prior(
161161
hsgp_dims = (dims[1], "m")
162162

163163
gp = pm.gp.HSGP(m=[hsgp_kwargs.m], L=[hsgp_kwargs.L], cov_func=cov_func)
164-
phi, sqrt_psd = gp.prior_linearized(Xs=X[:, None] - X_mid)
164+
phi, sqrt_psd = gp.prior_linearized(X[:, None] - X_mid)
165165
hsgp_coefs = pm.Normal(f"{name}_hsgp_coefs", dims=hsgp_dims)
166166
f = phi @ (hsgp_coefs * sqrt_psd).T
167167
f = pt.softplus(f)

0 commit comments

Comments
 (0)