Merge branch 'cleanup_src' of github.com:JaxGaussianProcesses/GPJax i…

…nto cleanup_src
JaxGaussianProcesses · Nov 28, 2023 · 54630e1 · 54630e1
2 parents 135b47f + af7bb11
commit 54630e1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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]