Add some additional methods to overrride the stored p and t

SciML · Nov 14, 2023 · c7096f2 · c7096f2
1 parent 44b8dd0
commit c7096f2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "SciMLBase"
 uuid = "0bca4576-84f4-4d90-8ffe-ffa030f20462"
 authors = ["Chris Rackauckas <[email protected]> and contributors"]
-version = "2.8.0"
+version = "2.8.1"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

diff --git a/src/function_wrappers.jl b/src/function_wrappers.jl
@@ -14,6 +14,8 @@ end
 
 (ff::UJacobianWrapper)(du1, uprev) = ff.f(du1, uprev, ff.p, ff.t)
 (ff::UJacobianWrapper)(uprev) = (du1 = similar(uprev); ff.f(du1, uprev, ff.p, ff.t); du1)
+(ff::UJacobianWrapper)(du1, uprev, p, t) = ff.f(du1, uprev, p, t)
+(ff::UJacobianWrapper)(uprev, p, t) = (du1 = similar(uprev); ff.f(du1, uprev, p, t); du1)
 
 mutable struct TimeDerivativeWrapper{F, uType, P} <: Function
     f::F