Skip to content

Commit 2161664

Browse files
Forward dims to reparametrized Deterministic
This way `.prior(..., dims=...)` applies not only to the `_rotated_` RV but also to the `Deterministic` that the user is interested in.
1 parent 9ca57bf commit 2161664

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pymc/gp/gp.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ def _build_prior(self, name, X, reparameterize=True, jitter=JITTER_DEFAULT, **kw
132132
if reparameterize:
133133
size = np.shape(X)[0]
134134
v = pm.Normal(name + "_rotated_", mu=0.0, sigma=1.0, size=size, **kwargs)
135-
f = pm.Deterministic(name, mu + cholesky(cov).dot(v))
135+
f = pm.Deterministic(name, mu + cholesky(cov).dot(v), dims=kwargs.get("dims", None))
136136
else:
137137
f = pm.MvNormal(name, mu=mu, cov=cov, **kwargs)
138138
return f
@@ -279,7 +279,7 @@ def _build_prior(self, name, X, reparameterize=True, jitter=JITTER_DEFAULT, **kw
279279
if reparameterize:
280280
size = np.shape(X)[0]
281281
v = pm.StudentT(name + "_rotated_", mu=0.0, sigma=1.0, nu=self.nu, size=size, **kwargs)
282-
f = pm.Deterministic(name, mu + cholesky(cov).dot(v))
282+
f = pm.Deterministic(name, mu + cholesky(cov).dot(v), dims=kwargs.get("dims", None))
283283
else:
284284
f = pm.MvStudentT(name, nu=self.nu, mu=mu, cov=cov, **kwargs)
285285
return f

0 commit comments

Comments
 (0)