Skip to content

Commit 5a2705e

Browse files
refactor: recursively check for parameter dependencies everywhere
Co-authored-by: Aayush Sabharwal <[email protected]>
1 parent 250e730 commit 5a2705e

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

src/systems/abstractsystem.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -970,7 +970,7 @@ function parameters(sys::AbstractSystem)
970970
result = unique(isempty(systems) ? ps :
971971
[ps; reduce(vcat, namespace_parameters.(systems))])
972972
if has_parameter_dependencies(sys) &&
973-
(pdeps = get_parameter_dependencies(sys)) !== nothing
973+
(pdeps = parameter_dependencies(sys)) !== nothing
974974
filter(result) do sym
975975
!haskey(pdeps, sym)
976976
end
@@ -2391,8 +2391,8 @@ function extend(sys::AbstractSystem, basesys::AbstractSystem; name::Symbol = nam
23912391
eqs = union(get_eqs(basesys), get_eqs(sys))
23922392
sts = union(get_unknowns(basesys), get_unknowns(sys))
23932393
ps = union(get_ps(basesys), get_ps(sys))
2394-
base_deps = get_parameter_dependencies(basesys)
2395-
deps = get_parameter_dependencies(sys)
2394+
base_deps = parameter_dependencies(basesys)
2395+
deps = parameter_dependencies(sys)
23962396
dep_ps = isnothing(base_deps) ? deps :
23972397
isnothing(deps) ? base_deps : union(base_deps, deps)
23982398
obs = union(get_observed(basesys), get_observed(sys))

src/systems/diffeqs/sdesystem.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -291,7 +291,7 @@ function stochastic_integral_transform(sys::SDESystem, correction_factor)
291291
end
292292

293293
SDESystem(deqs, get_noiseeqs(sys), get_iv(sys), unknowns(sys), parameters(sys),
294-
name = name, parameter_dependencies = get_parameter_dependencies(sys), checks = false)
294+
name = name, parameter_dependencies = parameter_dependencies(sys), checks = false)
295295
end
296296

297297
"""
@@ -399,7 +399,7 @@ function Girsanov_transform(sys::SDESystem, u; θ0 = 1.0)
399399
# return modified SDE System
400400
SDESystem(deqs, noiseeqs, get_iv(sys), unknown_vars, parameters(sys);
401401
defaults = Dict=> θ0), observed = [weight ~ θ / θ0],
402-
name = name, parameter_dependencies = get_parameter_dependencies(sys),
402+
name = name, parameter_dependencies = parameter_dependencies(sys),
403403
checks = false)
404404
end
405405

src/systems/parameter_buffer.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ function MTKParameters(
136136
nonnumeric_buffer = nonnumeric_buffer
137137

138138
if has_parameter_dependencies(sys) &&
139-
(pdeps = get_parameter_dependencies(sys)) !== nothing
139+
(pdeps = parameter_dependencies(sys)) !== nothing
140140
pdeps = Dict(k => fixpoint_sub(v, pdeps) for (k, v) in pdeps)
141141
dep_exprs = ArrayPartition((Any[missing for _ in 1:length(v)] for v in dep_buffer)...)
142142
for (sym, val) in pdeps

0 commit comments

Comments
 (0)