update test

TuringLang · Oct 28, 2024 · 3b05769 · 3b05769
1 parent 9d05e46
commit 3b05769
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/abstractmcmc.jl b/test/abstractmcmc.jl
@@ -26,10 +26,10 @@ using Statistics: mean
 
         θ = AbstractMCMC.getparams(s)
         @test θ == t.z.θ
-        @test AbstractMCMC.setparams!!(s, θ) == s
+        @test AbstractMCMC.setparams!!(model, s, θ) == s
 
         new_θ = randn(rng, 2)
-        new_state = AbstractMCMC.setparams!!(s, new_θ)
+        new_state = AbstractMCMC.setparams!!(model, s, new_θ)
         @test AbstractMCMC.getparams(new_state) == new_θ
     end