From af7bb115dfa47ea3151b01a5b748c7db41e45263 Mon Sep 17 00:00:00 2001
From: Daniel Dodd <daniel_dodd@icloud.com>
Date: Tue, 28 Nov 2023 18:13:59 +0000
Subject: [PATCH] Remove jit.

---
 docs/examples/graph_kernels.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docs/examples/graph_kernels.py b/docs/examples/graph_kernels.py
index 3afa2f1eb..795deb843 100644
--- a/docs/examples/graph_kernels.py
+++ b/docs/examples/graph_kernels.py
@@ -172,9 +172,9 @@
 # %%
 opt_posterior, training_history = gpx.fit_scipy(
     model=posterior,
-    objective=jit(gpx.ConjugateMLL(negative=True)),
+    objective=gpx.ConjugateMLL(negative=True),
     train_data=D,
-    max_iters=1000
+    max_iters=1000,
 )
 
 # %% [markdown]