fix typo

PFLeget · May 16, 2024 · a3d5eb8 · a3d5eb8
1 parent c1bb047
commit a3d5eb8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/treegp/gp_interp.py b/treegp/gp_interp.py
@@ -179,6 +179,8 @@ def return_gp_predict(self, y, X1, X2, kernel, y_err, return_cov=False):
             self._alpha = cho_solve(factor, y, overwrite_b=False)
         y_predict = np.dot(HT, self._alpha.reshape((len(self._alpha), 1))).T[0]
         if return_cov:
+            if self._alpha is not None:
+                K = kernel.__call__(X1) + np.eye(len(y)) * y_err**2
             fact = cholesky(
                 K, lower=True
             )  # I am computing maybe twice the same things...