Fix root_find(..., SomeLeastSquaresSolver(), ...) returning the wro…

…ng auxiliary structure.
patrick-kidger · Dec 23, 2024 · fa6ed87 · fa6ed87
1 parent 5c6335a
commit fa6ed87
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimistix/_root_find.py b/optimistix/_root_find.py
@@ -196,7 +196,7 @@ def root_find(
         return sol
     elif isinstance(solver, AbstractLeastSquaresSolver):
         del tags
-        return least_squares(
+        sol = least_squares(
             eqx.Partial(_to_lstsq_fn, fn),
             _LstsqToRoot(solver),  # pyright: ignore
             y0,