Skip to content

Commit cc9510c

Browse files
committed
Use ESS in Gibbs DEMO_MODELS tests
1 parent 583d297 commit cc9510c

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

test/mcmc/gibbs.jl

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -367,6 +367,7 @@ end
367367
samplers = [
368368
Turing.Gibbs(@varname(s) => NUTS(), @varname(m) => NUTS()),
369369
Turing.Gibbs(@varname(s) => NUTS(), @varname(m) => HMC(0.01, 4)),
370+
Turing.Gibbs(@varname(s) => NUTS(), @varname(m) => ESS()),
370371
]
371372

372373
if !has_dot_assume(model)
@@ -412,10 +413,7 @@ end
412413
initial_params = fill(initial_params, num_chains)
413414

414415
# Sampler to use for Gibbs components.
415-
sampler_inner = HMC(0.1, 32)
416-
sampler = Turing.Gibbs(
417-
@varname(s) => sampler_inner, @varname(m) => sampler_inner
418-
)
416+
sampler = Turing.Gibbs(@varname(s) => HMC(0.1, 32), @varname(m) => ESS())
419417
Random.seed!(42)
420418
chain = sample(
421419
model,

0 commit comments

Comments
 (0)