Fix

brentyi · Oct 12, 2024 · 2c41aef · 2c41aef
1 parent 7c9ecd1
commit 2c41aef
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/jaxls/_solvers.py b/src/jaxls/_solvers.py
@@ -91,7 +91,9 @@ class ConjugateGradientLinearSolver:
     For reference, see AN INEXACT LEVENBERG-MARQUARDT METHOD FOR LARGE SPARSE NONLINEAR
     LEAST SQUARES, Wright & Holt 1983."""
 
-    preconditioner: jdc.Static[Literal["block-jacobi", "point-jacobi"]] | None = "block-jacobi"
+    preconditioner: jdc.Static[Literal["block-jacobi", "point-jacobi"] | None] = (
+        "block-jacobi"
+    )
     """Preconditioner to use for linear solves."""
 
     def _solve(