try to fix doc example error

TuringLang · Oct 1, 2024 · c177271 · c177271
1 parent bceb510
commit c177271
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/docs/make.jl b/docs/make.jl
@@ -1,4 +1,7 @@
 using AbstractMCMC
+using AbstractPPL
+using BangBang
+using Distributions
 using Documenter
 using Random
 

diff --git a/docs/src/state_interface.md b/docs/src/state_interface.md
@@ -49,7 +49,6 @@ using Random # hide
 using AbstractMCMC: AbstractMCMC # hide 
 using AbstractPPL: AbstractPPL # hide
 using BangBang: constructorof # hide
-using AbstractPPL: AbstractPPL
 ```
 
 ```@example gibbs_example
@@ -493,11 +492,14 @@ samples = sample(
 Then we can extract the samples and compute the mean of the samples.
 
 ```@example gibbs_example
-mu_samples = [sample.values.mu for sample in samples][20001:end]
-tau2_samples = [sample.values.tau2 for sample in samples][20001:end]
-
-mean(mu_samples)
-mean(tau2_samples)
+warmup = 5000
+thin = 10
+thinned_samples = samples[(warmup + 1):thin:end]
+mu_samples = [sample.values.mu for sample in thinned_samples]
+tau2_samples = [sample.values.tau2 for sample in thinned_samples]
+
+mu_mean = only(mean(mu_samples))
+tau2_mean = only(mean(tau2_samples))
 (mu_mean, tau2_mean)
 ```