Skip to content

Commit

Permalink
fix: fix inconsistent defaults with default_toterm variables
Browse files Browse the repository at this point in the history
  • Loading branch information
AayushSabharwal committed Jul 4, 2024
1 parent 2cd9da1 commit 051a11e
Show file tree
Hide file tree
Showing 7 changed files with 22 additions and 1 deletion.
5 changes: 4 additions & 1 deletion src/systems/diffeqs/odesystem.jl
Original file line number Diff line number Diff line change
Expand Up @@ -238,6 +238,9 @@ function ODESystem(deqs::AbstractVector{<:Equation}, iv, dvs, ps;
end
defaults = todict(defaults)
defaults = Dict{Any, Any}(value(k) => value(v) for (k, v) in pairs(defaults))
for k in collect(keys(defaults))
defaults[default_toterm(k)] = defaults[k]
end
var_to_name = Dict()
process_variables!(var_to_name, defaults, dvs′)
process_variables!(var_to_name, defaults, ps′)
Expand Down Expand Up @@ -410,7 +413,7 @@ function build_explicit_observed_function(sys, ts;
Set(arguments(st)[1] for st in sts if iscall(st) && operation(st) === getindex))

observed_idx = Dict(x.lhs => i for (i, x) in enumerate(obs))
param_set = Set(full_parameters(sys))
param_set = Set(ps)
param_set = union(param_set,
Set(arguments(p)[1] for p in param_set if iscall(p) && operation(p) === getindex))
param_set_ns = Set(unknowns(sys, p) for p in full_parameters(sys))
Expand Down
3 changes: 3 additions & 0 deletions src/systems/diffeqs/sdesystem.jl
Original file line number Diff line number Diff line change
Expand Up @@ -187,6 +187,9 @@ function SDESystem(deqs::AbstractVector{<:Equation}, neqs::AbstractArray, iv, dv
end
defaults = todict(defaults)
defaults = Dict(value(k) => value(v) for (k, v) in pairs(defaults))
for k in collect(keys(defaults))
defaults[default_toterm(k)] = defaults[k]
end

var_to_name = Dict()
process_variables!(var_to_name, defaults, dvs′)
Expand Down
3 changes: 3 additions & 0 deletions src/systems/discrete_system/discrete_system.jl
Original file line number Diff line number Diff line change
Expand Up @@ -149,6 +149,9 @@ function DiscreteSystem(eqs::AbstractVector{<:Equation}, iv, dvs, ps;
end
defaults = todict(defaults)
defaults = Dict(value(k) => value(v) for (k, v) in pairs(defaults))
for k in collect(keys(defaults))
defaults[default_toterm(k)] = defaults[k]
end

var_to_name = Dict()
process_variables!(var_to_name, defaults, dvs′)
Expand Down
3 changes: 3 additions & 0 deletions src/systems/jumps/jumpsystem.jl
Original file line number Diff line number Diff line change
Expand Up @@ -175,6 +175,9 @@ function JumpSystem(eqs, iv, unknowns, ps;
end
defaults = todict(defaults)
defaults = Dict(value(k) => value(v) for (k, v) in pairs(defaults))
for k in collect(keys(defaults))
defaults[default_toterm(k)] = defaults[k]
end

unknowns, ps = value.(unknowns), value.(ps)
var_to_name = Dict()
Expand Down
3 changes: 3 additions & 0 deletions src/systems/nonlinear/nonlinearsystem.jl
Original file line number Diff line number Diff line change
Expand Up @@ -147,6 +147,9 @@ function NonlinearSystem(eqs, unknowns, ps;
jac = RefValue{Any}(EMPTY_JAC)
defaults = todict(defaults)
defaults = Dict{Any, Any}(value(k) => value(v) for (k, v) in pairs(defaults))
for k in collect(keys(defaults))
defaults[default_toterm(k)] = defaults[k]
end

unknowns, ps = value.(unknowns), value.(ps)
var_to_name = Dict()
Expand Down
3 changes: 3 additions & 0 deletions src/systems/optimization/constraints_system.jl
Original file line number Diff line number Diff line change
Expand Up @@ -134,6 +134,9 @@ function ConstraintsSystem(constraints, unknowns, ps;
jac = RefValue{Any}(EMPTY_JAC)
defaults = todict(defaults)
defaults = Dict(value(k) => value(v) for (k, v) in pairs(defaults))
for k in collect(keys(defaults))
defaults[default_toterm(k)] = defaults[k]
end

var_to_name = Dict()
process_variables!(var_to_name, defaults, unknowns′)
Expand Down
3 changes: 3 additions & 0 deletions src/systems/optimization/optimizationsystem.jl
Original file line number Diff line number Diff line change
Expand Up @@ -112,6 +112,9 @@ function OptimizationSystem(op, unknowns, ps;
end
defaults = todict(defaults)
defaults = Dict(value(k) => value(v) for (k, v) in pairs(defaults))
for k in collect(keys(defaults))
defaults[default_toterm(k)] = defaults[k]
end

var_to_name = Dict()
process_variables!(var_to_name, defaults, unknowns′)
Expand Down

0 comments on commit 051a11e

Please sign in to comment.