diff --git a/src/jaxls/_solvers.py b/src/jaxls/_solvers.py index 4b12482..b29a952 100644 --- 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(