Merge pull request #907 from AayushSabharwal/as/discprob-ctor

feat: add `constructorof` for `DiscreteProblem`
SciML · Jan 16, 2025 · 234dd64 · 234dd64
2 parents 7ed01df + c081f78
commit 234dd64
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/problems/discrete_problems.jl b/src/problems/discrete_problems.jl
@@ -122,6 +122,17 @@ struct DiscreteProblem{uType, tType, isinplace, P, F, K} <:
     end
 end
 
+function ConstructionBase.constructorof(::Type{P}) where {P <: DiscreteProblem}
+    function ctor(f, u0, tspan, p, kw)
+        if f isa AbstractDiscreteFunction
+            iip = isinplace(f)
+        else
+            iip = isinplace(f, 4)
+        end
+        return DiscreteProblem{iip}(f, u0, tspan, p; kw...)
+    end
+end
+
 """
     DiscreteProblem{isinplace}(f,u0,tspan,p=NullParameters(),callback=nothing)