fix more test error

TuringLang · Oct 31, 2024 · 30dca23 · 30dca23
1 parent 027cd31
commit 30dca23
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/test/abstractmcmc.jl b/test/abstractmcmc.jl
@@ -28,10 +28,14 @@ using Statistics: mean
         @test θ == t.z.θ
         new_state = AbstractMCMC.setparams!!(model, s, θ)
         @test new_state.transition.z.θ == θ
-        @test new_state.transition.z.ℓπ == s.transition.z.ℓπ
-        @test new_state.transition.z.ℓκ == s.transition.z.ℓκ
+        new_state_logπ = new_state.transition.z.ℓπ
+        @test new_state_logπ.value == s.transition.z.ℓπ.value
+        @test new_state_logπ.gradient == s.transition.z.ℓπ.gradient
+        new_state_logκ = new_state.transition.z.ℓκ
+        @test new_state_logκ.value == s.transition.z.ℓκ.value
+        @test new_state_logκ.gradient == s.transition.z.ℓκ.gradient
         @test new_state.transition.z.r == s.transition.z.r
-        
+
         new_θ = randn(rng, 2)
         new_state = AbstractMCMC.setparams!!(model, s, new_θ)
         @test AbstractMCMC.getparams(new_state) == new_θ