fixed incorrect call to acclogp_assume!!

TuringLang · Nov 28, 2024 · 545cfab · 545cfab
1 parent 361c45e
commit 545cfab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/context_implementations.jl b/src/context_implementations.jl
@@ -591,7 +591,7 @@ function get_and_set_val!(
             push!!.((vi,), vns, _link_broadcast_new.((vi,), vns, dists, r), dists, (spl,))
             # NOTE: Need to add the correction.
             # FIXME: This is not great.
-            acclogp_assume!!(vi, sum(logabsdetjac.(link_transform.(dists), r)))
+            acclogp!!(vi, sum(logabsdetjac.(link_transform.(dists), r)))
             # `push!!` sets the trans-flag to `false` by default.
             settrans!!.((vi,), true, vns)
         else