fixup! fix: fix @reset changing specialization of ODEFunction

SciML · Jul 22, 2024 · 5662ad2 · 5662ad2
1 parent b0e7699
commit 5662ad2
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/remake.jl b/src/remake.jl
@@ -126,10 +126,18 @@ function remake(prob::ODEProblem; f = missing,
         else
             _f = prob.f
             if __has_initializeprob(_f)
-                _f = parameterless_type(_f){iip, specialization(_f)}(; getproperties(_f)..., initializeprob)
+                props = getproperties(_f)
+                @reset props.initializeprob = initializeprob
+                props = values(props)
+                _f = parameterless_type(_f){
+                    iip, specialization(_f), map(typeof, props)...}(props...)
             end
             if __has_initializeprobmap(_f)
-                _f = parameterless_type(_f){iip, specialization(_f)}(; getproperties(_f)..., initializeprobmap)
+                props = getproperties(_f)
+                @reset props.initializeprobmap = initializeprobmap
+                props = values(props)
+                _f = parameterless_type(_f){
+                    iip, specialization(_f), map(typeof, props)...}(props...)
             end
         end
     elseif f isa AbstractODEFunction