Skip to content

Commit 6b61283

Browse files
committed
Add kwargs to modelingtoolkitize
1 parent 056e3c2 commit 6b61283

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

src/systems/diffeqs/modelingtoolkitize.jl

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ $(TYPEDSIGNATURES)
33
44
Generate `ODESystem`, dependent variables, and parameters from an `ODEProblem`.
55
"""
6-
function modelingtoolkitize(prob::DiffEqBase.ODEProblem)
6+
function modelingtoolkitize(prob::DiffEqBase.ODEProblem; kwargs...)
77
prob.f isa DiffEqBase.AbstractParameterizedFunction &&
88
return prob.f.sys
99
@parameters t
@@ -65,7 +65,8 @@ function modelingtoolkitize(prob::DiffEqBase.ODEProblem)
6565

6666
de = ODESystem(
6767
eqs, t, sts, params,
68-
defaults=merge(default_u0, default_p),
68+
defaults=merge(default_u0, default_p);
69+
kwargs...
6970
)
7071

7172
de
@@ -96,7 +97,7 @@ $(TYPEDSIGNATURES)
9697
9798
Generate `SDESystem`, dependent variables, and parameters from an `SDEProblem`.
9899
"""
99-
function modelingtoolkitize(prob::DiffEqBase.SDEProblem)
100+
function modelingtoolkitize(prob::DiffEqBase.SDEProblem; kwargs...)
100101
prob.f isa DiffEqBase.AbstractParameterizedFunction &&
101102
return (prob.f.sys, prob.f.sys.states, prob.f.sys.ps)
102103
@parameters t
@@ -142,7 +143,7 @@ function modelingtoolkitize(prob::DiffEqBase.SDEProblem)
142143
Vector(vec(params))
143144
end
144145

145-
de = SDESystem(deqs,neqs,t,Vector(vec(vars)),params)
146+
de = SDESystem(deqs,neqs,t,Vector(vec(vars)),params; kwargs...)
146147

147148
de
148149
end
@@ -153,7 +154,7 @@ $(TYPEDSIGNATURES)
153154
154155
Generate `OptimizationSystem`, dependent variables, and parameters from an `OptimizationProblem`.
155156
"""
156-
function modelingtoolkitize(prob::DiffEqBase.OptimizationProblem)
157+
function modelingtoolkitize(prob::DiffEqBase.OptimizationProblem; kwargs...)
157158

158159
if prob.p isa Tuple || prob.p isa NamedTuple
159160
p = [x for x in prob.p]
@@ -166,6 +167,6 @@ function modelingtoolkitize(prob::DiffEqBase.OptimizationProblem)
166167
reshape([Num(Sym{Real}(nameof(Variable(, i)))) for i in eachindex(p)],size(Array(p)))
167168

168169
eqs = prob.f(vars, params)
169-
de = OptimizationSystem(eqs,vec(vars),vec(params))
170+
de = OptimizationSystem(eqs,vec(vars),vec(params); kwargs...)
170171
de
171172
end

0 commit comments

Comments
 (0)