Skip to content

Commit d60306f

Browse files
fix tests
1 parent ca8c538 commit d60306f

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

src/systems/diffeqs/diffeqsystem.jl

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -132,9 +132,13 @@ function generate_ode_iW(sys::DiffEqSystem,simplify=true)
132132
:((iW,u,p,gam,t)->$(block)),:((iW,u,p,gam,t)->$(block2))
133133
end
134134

135-
function DiffEqBase.ODEFunction(sys::DiffEqSystem)
136-
expr = generate_ode_function(sys)
137-
ODEFunction{true}(eval(expr))
135+
function DiffEqBase.ODEFunction(sys::DiffEqSystem;version = ArrayFunction,kwargs...)
136+
expr = generate_ode_function(sys;kwargs...)
137+
if version == ArrayFunction
138+
ODEFunction{true}(eval(expr))
139+
elseif version == SArrayFunction
140+
ODEFunction{false}(eval(expr))
141+
end
138142
end
139143

140144
export DiffEqSystem, ODEFunction

0 commit comments

Comments
 (0)