Add indepsym where missing

Author: AayushSabharwal

src/systems/diffeqs/abstractodesystem.jl

@@ -450,10 +450,9 @@ function DiffEqBase.DAEFunction{iip}(sys::AbstractODESystem, dvs = states(sys),
                      sys = sys,
                      jac = _jac === nothing ? nothing : _jac,
                      syms = Symbol.(dvs),
+                     indepsym = Symbol(get_iv(sys)),
                      paramsyms = Symbol.(ps),
                      jac_prototype = jac_prototype,
-                     # missing fields in `DAEFunction`
-                     #indepsym = Symbol(get_iv(sys)),
                      observed = observedfun)
 end
 
@@ -536,7 +535,8 @@ function ODEFunctionExpr{iip}(sys::AbstractODESystem, dvs = states(sys),
                           mass_matrix = M,
                           jac_prototype = $jp_expr,
                           syms = $(Symbol.(states(sys))),
-                          indepsym = $(QuoteNode(Symbol(get_iv(sys)))))
+                          indepsym = $(QuoteNode(Symbol(get_iv(sys)))),
+                          paramsyms = $(QuoteNode(Symbol.(parameters(sys)))))
     end
     !linenumbers ? striplines(ex) : ex
 end

src/systems/diffeqs/sdesystem.jl

@@ -421,6 +421,7 @@ function DiffEqBase.SDEFunction{iip}(sys::SDESystem, dvs = states(sys),
                      Wfact_t = _Wfact_t === nothing ? nothing : _Wfact_t,
                      mass_matrix = _M,
                      syms = Symbol.(states(sys)),
+                     indepsym = Symbol(get_iv(sys)),
                      paramsyms = Symbol.(ps),
                      observed = observedfun)
 end
@@ -507,6 +508,7 @@ function SDEFunctionExpr{iip}(sys::SDESystem, dvs = states(sys),
                           Wfact_t = Wfact_t,
                           mass_matrix = M,
                           syms = $(Symbol.(states(sys))),
+                          indepsym = $(Symbol(get_iv(sys))),
                           paramsyms = $(Symbol.(parameters(sys))))
     end
     !linenumbers ? striplines(ex) : ex

src/systems/discrete_system/discrete_system.jl

@@ -206,7 +206,7 @@ function SciMLBase.DiscreteProblem(sys::DiscreteSystem, u0map, tspan,
                               expression_module = eval_module)
     f_oop, _ = (@RuntimeGeneratedFunction(eval_module, ex) for ex in f_gen)
     f(u, p, iv) = f_oop(u, p, iv)
-    fd = DiscreteFunction(f; syms = Symbol.(dvs), paramsyms = Symbol.(ps), sys = sys)
+    fd = DiscreteFunction(f; syms = Symbol.(dvs), indepsym = Symbol(iv), paramsyms = Symbol.(ps), sys = sys)
     DiscreteProblem(fd, u0, tspan, p; kwargs...)
 end
 

src/systems/jumps/jumpsystem.jl

@@ -292,6 +292,7 @@ function DiffEqBase.DiscreteProblem(sys::JumpSystem, u0map, tspan::Union{Tuple,
     end
 
     df = DiscreteFunction{true, true}(f; syms = Symbol.(states(sys)),
+                                      indepsym = Symbol(get_iv(sys)),
                                       paramsyms = Symbol.(ps), sys = sys,
                                       observed = observedfun)
     DiscreteProblem(df, u0, tspan, p; kwargs...)
@@ -333,6 +334,7 @@ function DiscreteProblemExpr(sys::JumpSystem, u0map, tspan::Union{Tuple, Nothing
         p = $p
         tspan = $tspan
         df = DiscreteFunction{true, true}(f, syms = $(Symbol.(states(sys))),
+                                          indepsym = $(Symbol(get_iv(sys))),
                                           paramsyms = $(Symbol.(parameters(sys))))
         DiscreteProblem(df, u0, tspan, p)
     end