feat: add constructorof for DiscreteProblem

SciML · Jan 16, 2025 · c081f78 · c081f78
1 parent 7ed01df
commit c081f78
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)