Explicitly convert the type of values before create a Chain (#97)

Project.toml

@@ -1,6 +1,6 @@
 name = "JuliaBUGS"
 uuid = "ba9fb4c0-828e-4473-b6a1-cd2560fee5bf"
-version = "0.2.0"
+version = "0.2.1"
 
 [deps]
 AbstractPPL = "7a57a42e-76ec-4ea3-a279-07e840d6d9cf"

ext/JuliaBUGSAdvancedHMCExt.jl

@@ -40,8 +40,8 @@ function AbstractMCMC.bundle_samples(
     generated_quantities = []
     for i in eachindex(ts)
         vi = first(evaluate!!(model, LogDensityContext(), samples[i]))
-        push!(param_vals, vi[param_vars])
-        push!(generated_quantities, vi[generated_vars])
+        push!(param_vals, [vi[param_var] for param_var in param_vars])
+        push!(generated_quantities, [vi[generated_var] for generated_var in generated_vars])
     end
 
     param_name_leaves = vcat(
@@ -61,11 +61,14 @@ function AbstractMCMC.bundle_samples(
     flattened_param_vals = [vcat(p...) for p in param_vals]
     flattened_generated_quantities = [vcat(gq...) for gq in generated_quantities]
     vals = [
-        vcat(
-            flattened_param_vals[i],
-            flattened_generated_quantities[i],
-            ts[i].z.ℓπ.value,
-            collect(values(AdvancedHMC.stat(ts[i]))),
+        convert(
+            Vector{Real},
+            vcat(
+                flattened_param_vals[i],
+                flattened_generated_quantities[i],
+                ts[i].z.ℓπ.value,
+                collect(values(AdvancedHMC.stat(ts[i]))),
+            ),
         ) for i in eachindex(ts)
     ]
 

src/graphs.jl

@@ -35,6 +35,25 @@ struct ConcreteNodeInfo <: NodeInfo
     node_args::Vector
 end
 
+function Base.show(io::IO, n::ConcreteNodeInfo)
+    if n isa ConcreteNodeInfo
+        print(
+            io,
+            "ConcreteNodeInfo(\n",
+            "\tNode Type: ",
+            n.node_type,
+            "\n",
+            "\tNode Function Expression: ",
+            n.node_function_expr,
+            "\n",
+            "\tNode Arguments: ",
+            n.node_args,
+            "\n",
+            ")",
+        )
+    end
+end
+
 function ConcreteNodeInfo(var::Var, vars, node_functions, node_args)
     return ConcreteNodeInfo(
         vars[var],