Use a mutable struct instead of Dict for Safe termination

avik-pal · avik-pal · commit 4bc41c760715 · 2023-04-13T12:51:40.000-04:00
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,3 @@
 Manifest.toml
-
+.vscode
 wip
diff --git a/Project.toml b/Project.toml
@@ -23,7 +23,7 @@ SimpleBatchedNonlinearSolveExt = "NNlib"
 
 [compat]
 ArrayInterface = "6, 7"
-DiffEqBase = "6.119"
+DiffEqBase = "6.123.0"
 FiniteDiff = "2"
 ForwardDiff = "0.10.3"
 NNlib = "0.8"
diff --git a/ext/SimpleBatchedNonlinearSolveExt.jl b/ext/SimpleBatchedNonlinearSolveExt.jl
@@ -59,7 +59,8 @@ function SciMLBase.__solve(prob::NonlinearProblem, alg::Broyden{true}, args...;
         error("Broyden currently doesn't support SAFE_BEST termination modes")
     end
 
-    storage = mode ∈ DiffEqBase.SAFE_TERMINATION_MODES ? Dict() : nothing
+    storage = mode ∈ DiffEqBase.SAFE_TERMINATION_MODES ? NLSolveSafeTerminationResult() :
+              nothing
     termination_condition = tc(storage)
 
     xₙ = x
diff --git a/src/broyden.jl b/src/broyden.jl
@@ -48,7 +48,8 @@ function SciMLBase.__solve(prob::NonlinearProblem, alg::Broyden{false}, args...;
         error("Broyden currently doesn't support SAFE_BEST termination modes")
     end
 
-    storage = mode ∈ DiffEqBase.SAFE_TERMINATION_MODES ? Dict() : nothing
+    storage = mode ∈ DiffEqBase.SAFE_TERMINATION_MODES ? NLSolveSafeTerminationResult() :
+              nothing
     termination_condition = tc(storage)
 
     xₙ = x
diff --git a/src/lbroyden.jl b/src/lbroyden.jl
@@ -63,7 +63,8 @@ end
         error("LBroyden currently doesn't support SAFE_BEST termination modes")
     end
 
-    storage = mode ∈ DiffEqBase.SAFE_TERMINATION_MODES ? Dict() : nothing
+    storage = mode ∈ DiffEqBase.SAFE_TERMINATION_MODES ? NLSolveSafeTerminationResult() :
+              nothing
     termination_condition = tc(storage)
 
     xₙ = x

-Original file line number
+Diff line change
@@ @@ -1,3 +1,3 @@ @@
 Manifest.toml
+-
 +.vscode
 wip