fixed

JaxGaussianProcesses · Sep 22, 2023 · 57f0dc3 · 57f0dc3
1 parent 62c889f
commit 57f0dc3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gpjax/fit.py b/gpjax/fit.py
@@ -131,7 +131,8 @@ def evaluate(self, model: LinearModel, train_data: gpx.Dataset) -> float:
 
     # Initialise solver state.
     solver.fun = _wrap_objective(solver.fun)
-    solver.options.pop("maxiter", None)  # allow __post_init__ without jaxopt error
+    if hasattr(solver, "options"):  # allow __post_init__ without weird jaxopt error
+        solver.options.pop("maxiter", None)
     solver.__post_init__()  # needed to propagate changes to `fun` attribute
 
     if isinstance(solver, OptaxSolver):  # hack for Optax compatibility