diff --git a/test/turing/varinfo.jl b/test/turing/varinfo.jl index 8aa19241d..e5b8eb79f 100644 --- a/test/turing/varinfo.jl +++ b/test/turing/varinfo.jl @@ -348,10 +348,10 @@ @model function f(x) ns ~ filldist(Normal(0, 2.0), 3) m ~ Uniform(0, 1) - x ~ Normal(m, 1) + return x ~ Normal(m, 1) end model = f(1) - chain = sample(model, NUTS(), MCMCThreads(), 10, 2); + chain = sample(model, NUTS(), MCMCThreads(), 10, 2) loglikelihood(model, chain) logprior(model, chain) logjoint(model, chain)