Fix VarName parameters.

TuringLang · Jul 25, 2017 · b4ef284 · b4ef284
1 parent a441583
commit b4ef284
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/core/compiler.jl b/src/core/compiler.jl
@@ -209,13 +209,15 @@ macro model(fexpr)
   return_ex = fbody.args[end]   # get last statement of defined model
   if typeof(return_ex) == Symbol
     pop!(fbody_inner.args)
-    push!(fbody_inner.args, :(vn = Turing.VarName(:ret, $return_ex, "", 1)))
+    vstr = string(return_ex)
+    push!(fbody_inner.args, :(vn = Turing.VarName(:ret, Symbol($vstr), "", 1)))
     push!(fbody_inner.args, Expr(:(=), Expr(:ref, :vi, :vn), return_ex))
   elseif return_ex.head == :return || return_ex.head == :tuple
     pop!(fbody_inner.args)
     for v = return_ex.args
       @assert typeof(v) == Symbol "Returned variable name must be a symbol."
-      push!(fbody_inner.args, :(vn = Turing.VarName(:ret, $v, "", 1)))
+      vstr = string(v)
+      push!(fbody_inner.args, :(vn = Turing.VarName(:ret, Symbol($vstr), "", 1)))
       push!(fbody_inner.args, Expr(:(=), Expr(:ref, :vi, :vn), v))
     end
   end