Bugfix and minor tweaks of test settings.

TuringLang · Jul 31, 2023 · 8bdb40c · 8bdb40c
1 parent 4f6fde8
commit 8bdb40c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/inference/gibbs.jl b/src/inference/gibbs.jl
@@ -67,8 +67,7 @@ function Gibbs(
     arg1::Union{Tuple{<:InferenceAlgorithm,Int}, Pair{<:InferenceAlgorithm,Int}}, 
     argrest::Vararg{<:Union{Tuple{<:InferenceAlgorithm,Int}, Pair{<:InferenceAlgorithm,Int}}, N},
 ) where {N}
-    args = (arg1, argrest...)
-    allargs = (arg, args...)
+    allargs = (arg1, argrest...)
     algs = map(first, allargs)
     iterations = map(last, allargs)
     # obtain space of sampling algorithms

diff --git a/test/inference/gibbs.jl b/test/inference/gibbs.jl
@@ -55,12 +55,13 @@
         chain = sample(MoGtest_default, gibbs, 10_000)
         check_MoGtest_default(chain, atol=0.15)
 
+        Random.seed!(200)
         for alg in [
-            Gibbs((MH(:s), 10), (HMC(0.2, 4, :m), 10)),
-            Gibbs(MH(:s) => 10, HMC(0.2, 4, :m) => 10),
+            Gibbs((MH(:s), 2), (HMC(0.2, 4, :m), 1)),
+            Gibbs(MH(:s) => 1, HMC(0.2, 4, :m) => 2),
         ]
-            chain = sample(gdemo(1.5, 2.0), alg, 5_000)
-            check_gdemo(chain; atol=0.1)
+            chain = sample(gdemo(1.5, 2.0), alg, 10_000)
+            check_gdemo(chain; atol=0.15)
         end
     end