diff --git a/src/blox/discrete.jl b/src/blox/discrete.jl index 1e3aca72..e6b5ee9c 100644 --- a/src/blox/discrete.jl +++ b/src/blox/discrete.jl @@ -59,7 +59,7 @@ struct TAN <: AbstractDiscrete function TAN(; name, namespace=nothing, κ=100, λ=1) sts = @variables R(t) - ps = @parameters κ=κ λ=λ jcn=0 [input=true] + ps = @parameters κ=κ λ=λ jcn [input=true] eqs = [ R ~ min(κ, κ/(λ*jcn + sqrt(eps()))) ] @@ -79,7 +79,7 @@ struct SNc <: AbstractModulator function SNc(; name, namespace=nothing, κ_DA=1, N_time_blocks=5, DA_reward=10, λ_DA=0.33, t_event=90.0) sts = @variables R(t) R_(t) - ps = @parameters κ=κ_DA λ=λ_DA jcn=0 [input=true] jcn_=0 #HACK: jcn_ stores the value of jcn at time t_event that can be accessed after the simulation + ps = @parameters κ=κ_DA λ=λ_DA jcn [input=true] jcn_=0 #HACK: jcn_ stores the value of jcn at time t_event that can be accessed after the simulation eqs = [ R ~ min(κ, κ/(λ*jcn + sqrt(eps()))),