Skip to content

Commit 1c9764e

Browse files
shashiYingboMa
andcommitted
fixes
Co-authored-by: "Yingbo Ma" <[email protected]>
1 parent 79cb104 commit 1c9764e

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

src/systems/diffeqs/sdesystem.jl

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -166,6 +166,9 @@ function DiffEqBase.SDEFunction{iip}(sys::SDESystem, dvs = states(sys), ps = par
166166
u0 = nothing;
167167
version = nothing, tgrad=false, sparse = false,
168168
jac = false, Wfact = false, eval_expression = true, kwargs...) where {iip}
169+
dvs = scalarize.(dvs)
170+
ps = scalarize.(ps)
171+
169172
f_gen = generate_function(sys, dvs, ps; expression=Val{eval_expression}, kwargs...)
170173
f_oop,f_iip = eval_expression ? (@RuntimeGeneratedFunction(ex) for ex in f_gen) : f_gen
171174
g_gen = generate_diffusion_function(sys, dvs, ps; expression=Val{eval_expression}, kwargs...)

src/systems/reaction/reactionsystem.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -461,11 +461,13 @@ This input may contain repeat parameters.
461461
function Base.convert(::Type{<:SDESystem}, rs::ReactionSystem;
462462
noise_scaling=nothing, name=nameof(rs), combinatoric_ratelaws=true, kwargs...)
463463

464-
if noise_scaling isa Vector
464+
if noise_scaling isa AbstractArray
465465
(length(noise_scaling)!=length(equations(rs))) &&
466466
error("The number of elements in 'noise_scaling' must be equal " *
467467
"to the number of reactions in the reaction system.")
468-
noise_scaling = value.(noise_scaling)
468+
if !(noise_scaling isa Symbolics.Arr)
469+
noise_scaling = value.(noise_scaling)
470+
end
469471
elseif !isnothing(noise_scaling)
470472
noise_scaling = fill(value(noise_scaling),length(equations(rs)))
471473
end

0 commit comments

Comments
 (0)