Use median instead of mean in a test to lower the variance of the

estimate a bit
TuringLang · Sep 26, 2024 · 03e84b3 · 03e84b3
1 parent 4088a9f
commit 03e84b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/abstractmcmc.jl b/test/abstractmcmc.jl
@@ -222,7 +222,7 @@
 
         # Check that means are roughly okay.
         params_resolved = map(first ∘ MCMCTempering.getparams, transitions_resolved_inner)
-        @test vec(mean(params_resolved; dims=2)) ≈ [5.0, 5.0] atol = 0.4
+        @test vec(median(params_resolved; dims=2)) ≈ [5.0, 5.0] atol = 0.4
 
         # A composition of `SwapSampler` and `MultiSampler` has special `AbstractMCMC.bundle_samples`.
         @testset "bundle_samples with Vector" begin