Merge pull request #740 from AayushSabharwal/as/fix-remake-specializa…

…tion fix: fix `remake` changing specialization of `ODEFunction`
SciML · Jul 22, 2024 · 12b8832 · 12b8832
2 parents 9978e5b + d2c7570
commit 12b8832
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/src/remake.jl b/src/remake.jl
@@ -125,8 +125,20 @@ function remake(prob::ODEProblem; f = missing,
             end
         else
             _f = prob.f
-            @reset _f.initializeprob = initializeprob
-            @reset _f.initializeprobmap = initializeprobmap
+            if __has_initializeprob(_f)
+                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)
+                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
         _f = f