From 9049189bef63d0345bd94f99c09d6ca0491ef323 Mon Sep 17 00:00:00 2001
From: henrymoss <32096840+henrymoss@users.noreply.github.com>
Date: Mon, 17 Jun 2024 11:53:41 +0100
Subject: [PATCH] fixed

---
 gpjax/gps.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gpjax/gps.py b/gpjax/gps.py
index f5fb80475..9ba3aa097 100644
--- a/gpjax/gps.py
+++ b/gpjax/gps.py
@@ -532,7 +532,7 @@ def predict(
         mean_t = self.prior.mean_function(t)
         Ktt = self.prior.kernel.gram(t)
         Kxt = self.prior.kernel.cross_covariance(x, t)
-        Sigma_inv_Kxt = cola.solve(Sigma, Kxt)
+        Sigma_inv_Kxt = cola.solve(Sigma, Kxt, Cholesky())
 
         # μt  +  Ktx (Kxx + Io²)⁻¹ (y  -  μx)
         mean = mean_t + jnp.matmul(Sigma_inv_Kxt.T, y - mx)