Skip to content

Commit d17e6ed

Browse files
committed
Fix RepeatSampler in Gibbs bug
1 parent 1041449 commit d17e6ed

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/mcmc/gibbs.jl

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -292,6 +292,9 @@ wrap_algorithm_maybe(x) = x
292292
function wrap_algorithm_maybe(x::DynamicPPL.Sampler)
293293
return DynamicPPL.Sampler(x.alg, DynamicPPL.Selector(0))
294294
end
295+
function wrap_algorithm_maybe(x::RepeatSampler)
296+
return RepeatSampler(wrap_algorithm_maybe(x.sampler), x.num_repeat)
297+
end
295298
wrap_algorithm_maybe(x::InferenceAlgorithm) = DynamicPPL.Sampler(x, DynamicPPL.Selector(0))
296299

297300
"""

0 commit comments

Comments
 (0)