From 874c09e61f8720ae40ef6f8855cd05d499c0d8bb Mon Sep 17 00:00:00 2001
From: Avik Pal <avikpal@mit.edu>
Date: Wed, 15 Nov 2023 20:21:43 -0500
Subject: [PATCH 1/2] Use Function Wrappers from SciMLBase

---
 Project.toml      |  2 +-
 src/jacobian.jl   | 15 ++-------------
 src/linesearch.jl |  2 +-
 3 files changed, 4 insertions(+), 15 deletions(-)

diff --git a/Project.toml b/Project.toml
index fcc210bef..0eb72ea4e 100644
--- a/Project.toml
+++ b/Project.toml
@@ -60,7 +60,7 @@ SciMLBase = "2.4"
 SciMLOperators = "0.3"
 SimpleNonlinearSolve = "0.1.23"
 SparseArrays = "1.9"
-SparseDiffTools = "2.11"
+SparseDiffTools = "2.12"
 StaticArraysCore = "1.4"
 UnPack = "1.0"
 Zygote = "0.6"
diff --git a/src/jacobian.jl b/src/jacobian.jl
index 368e0bb70..ac824559b 100644
--- a/src/jacobian.jl
+++ b/src/jacobian.jl
@@ -1,14 +1,3 @@
-@concrete struct JacobianWrapper{iip} <: Function
-    f
-    p
-end
-
-# Previous Implementation did not hold onto `iip`, but this causes problems in packages
-# where we check for the presence of function signatures to check which dispatch to call
-(uf::JacobianWrapper{false})(u) = uf.f(u, uf.p)
-(uf::JacobianWrapper{false})(res, u) = (vec(res) .= vec(uf.f(u, uf.p)))
-(uf::JacobianWrapper{true})(res, u) = uf.f(res, u, uf.p)
-
 sparsity_detection_alg(_, _) = NoSparsityDetection()
 function sparsity_detection_alg(f, ad::AbstractSparseADType)
     if f.sparsity === nothing
@@ -52,7 +41,7 @@ jacobian!!(::Number, cache) = last(value_derivative(cache.uf, cache.u))
 function jacobian_caches(alg::AbstractNonlinearSolveAlgorithm, f::F, u, p, ::Val{iip};
         linsolve_kwargs = (;), lininit::Val{linsolve_init} = Val(true),
         linsolve_with_JᵀJ::Val{needsJᵀJ} = Val(false)) where {iip, needsJᵀJ, linsolve_init, F}
-    uf = JacobianWrapper{iip}(f, p)
+    uf = SciMLBase.JacobianWrapper{iip}(f, p)
 
     haslinsolve = hasfield(typeof(alg), :linsolve)
 
@@ -152,7 +141,7 @@ function jacobian_caches(alg::AbstractNonlinearSolveAlgorithm, f::F, u::Number,
         ::Val{false}; linsolve_with_JᵀJ::Val{needsJᵀJ} = Val(false),
         kwargs...) where {needsJᵀJ, F}
     # NOTE: Scalar `u` assumes scalar output from `f`
-    uf = JacobianWrapper{false}(f, p)
+    uf = SciMLBase.JacobianWrapper{false}(f, p)
     needsJᵀJ && return uf, nothing, u, nothing, nothing, u, u, u
     return uf, nothing, u, nothing, nothing, u
 end
diff --git a/src/linesearch.jl b/src/linesearch.jl
index a2b396b06..598934d03 100644
--- a/src/linesearch.jl
+++ b/src/linesearch.jl
@@ -122,7 +122,7 @@ function LineSearchesJLCache(ls::LineSearch, f::F, u, p, fu1, IIP::Val{iip}) whe
     end
 
     function g!(u, fu)
-        op = VecJac(f, u, p; fu = fu1, autodiff)
+        op = VecJac(SciMLBase.JacobianWrapper(f, p), u; fu = fu1, autodiff)
         if iip
             mul!(g₀, op, fu)
             return g₀

From af40e41047a79f5a3d6563ff27022aa8358953af Mon Sep 17 00:00:00 2001
From: Avik Pal <avikpal@mit.edu>
Date: Fri, 17 Nov 2023 09:57:01 -0500
Subject: [PATCH 2/2] Bump project version

---
 Project.toml                          | 4 ++--
 docs/src/tutorials/getting_started.md | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/Project.toml b/Project.toml
index 0eb72ea4e..2d8e4b661 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "NonlinearSolve"
 uuid = "8913a72c-1f9b-4ce2-8d82-65094dcecaec"
 authors = ["SciML"]
-version = "2.8.1"
+version = "2.8.2"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
@@ -56,7 +56,7 @@ NonlinearProblemLibrary = "0.1"
 PrecompileTools = "1"
 RecursiveArrayTools = "2"
 Reexport = "0.2, 1"
-SciMLBase = "2.4"
+SciMLBase = "2.8.2"
 SciMLOperators = "0.3"
 SimpleNonlinearSolve = "0.1.23"
 SparseArrays = "1.9"
diff --git a/docs/src/tutorials/getting_started.md b/docs/src/tutorials/getting_started.md
index 13ae48029..a5cc8786e 100644
--- a/docs/src/tutorials/getting_started.md
+++ b/docs/src/tutorials/getting_started.md
@@ -77,7 +77,7 @@ There are multiple return codes which can mean the solve was successful, and thu
 general command `SciMLBase.successful_retcode` to check whether the solution process exited as
 intended:
 
-```@example
+```@example 1
 SciMLBase.successful_retcode(sol)
 ```