Skip to content

Commit e8ce688

Browse files
HasnainRaztwiecki
authored andcommitted
Fix pymc module usage in KL
1 parent 40e28e8 commit e8ce688

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc_experimental/gp/latent_approx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ def _build_conditional(self, Xnew, X, f, U, s, jitter):
127127
Kxxpinv = U @ pt.diag(1.0 / s) @ U.T
128128
mus = Kxs.T @ Kxxpinv @ f
129129
K = Kss - Kxs.T @ Kxxpinv @ Kxs
130-
L = pm.gp.util.cholesky(pm.gp.util.stabilize(K, jitter))
130+
L = cholesky(stabilize(K, jitter))
131131
return mus, L
132132

133133
def conditional(self, name, Xnew, jitter=1e-6, **kwargs):

0 commit comments

Comments
 (0)