Update stan_inference.jl

SciML · Apr 12, 2020 · efa09e8 · efa09e8
1 parent f7b6f70
commit efa09e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/stan_inference.jl b/src/stan_inference.jl
@@ -123,7 +123,7 @@ function stan_inference(prob::DiffEqBase.DEProblem,t,data,priors = nothing;alg=:
   """
   if isnothing(diffeq_string)
     diffeq_string = ModelingToolkit.build_function(
-        sys.eqs,sys.dvs,
+        sys.eqs,sys.states,
         sys.ps,sys.iv,
         fname = :sho,
         target = ModelingToolkit.StanTarget()
@@ -167,4 +167,4 @@ function stan_inference(prob::DiffEqBase.DEProblem,t,data,priors = nothing;alg=:
   parameter_estimation_data = Dict("u0"=>prob.u0, "T" => length(t), "internal_var___u" => view(data, :, 1:length(t))', "t0" => prob.tspan[1], "ts" => t)
   return_code, chains, cnames = CmdStan.stan(stanmodel, [parameter_estimation_data]; CmdStanDir=CMDSTAN_HOME)
   return StanModel(stanmodel, return_code, chains, cnames)
-end
+end