Skip to content

Commit 82d52ef

Browse files
Fix OptimizationSystemExpr construction
1 parent e5e55c1 commit 82d52ef

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

src/systems/optimization/optimizationsystem.jl

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -90,10 +90,6 @@ struct AutoModelingToolkit <: DiffEqBase.AbstractADType end
9090
DiffEqBase.OptimizationProblem(sys::OptimizationSystem,args...;kwargs...) =
9191
DiffEqBase.OptimizationProblem{true}(sys::OptimizationSystem,args...;kwargs...)
9292

93-
OptimizationProblemExpr(sys::OptimizationSystem,args...;kwargs...) =
94-
OptimizationProblemExpr{true}(sys::OptimizationSystem,args...;kwargs...)
95-
96-
9793
"""
9894
```julia
9995
function DiffEqBase.OptimizationProblem{iip}(sys::OptimizationSystem,
@@ -168,6 +164,9 @@ calculating numerical enhancements.
168164
"""
169165
struct OptimizationProblemExpr{iip} end
170166

167+
OptimizationProblemExpr(sys::OptimizationSystem,args...;kwargs...) =
168+
OptimizationProblemExpr{true}(sys::OptimizationSystem,args...;kwargs...)
169+
171170
function OptimizationProblemExpr{iip}(sys::OptimizationSystem, u0,
172171
parammap=DiffEqBase.NullParameters();
173172
lb=nothing, ub=nothing,

test/controlsystem.jl

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,4 +13,7 @@ eqs = [
1313
sys = ControlSystem(loss,eqs,t,[x,v],[u],[p])
1414
dt = 0.1
1515
tspan = (0.0,1.0)
16-
runge_kutta_discretize(sys,dt,tspan)
16+
sys = runge_kutta_discretize(sys,dt,tspan)
17+
18+
u0 = rand(112) # guess for the state values
19+
prob = OptimizationProblem(sys,u0,[0.1],grad=true)

0 commit comments

Comments
 (0)