Skip to content

Commit 4991b6a

Browse files
committed
Fix HSGP docstrings examples
1 parent a8e892b commit 4991b6a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pymc/gp/hsgp_approx.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -446,7 +446,7 @@ class HSGPPeriodic(Base):
446446
447447
with pm.Model() as model:
448448
# Specify the covariance function, only for the 1-D case
449-
scale = pm.HalfNormal(10)
449+
scale = pm.HalfNormal("scale", 10)
450450
cov_func = pm.gp.cov.Periodic(1, period=1, ls=0.1)
451451
452452
# Specify the approximation with 25 basis vectors
@@ -535,8 +535,8 @@ def prior_linearized(self, Xs: TensorLike):
535535
X = np.linspace(0, 10, 100)[:, None]
536536
537537
with pm.Model() as model:
538-
scale = pm.HalfNormal(10)
539-
cov_func = pm.gp.cov.Periodic(1, period=1, ls=ell)
538+
scale = pm.HalfNormal("scale", 10)
539+
cov_func = pm.gp.cov.Periodic(1, period=1.0, ls=2.0)
540540
541541
# m=200 means 200 basis vectors
542542
gp = pm.gp.HSGPPeriodic(m=200, scale=scale, cov_func=cov_func)

0 commit comments

Comments
 (0)