Implement tilde_assume!! for Sampleable / pointwise contexts

Author: penelopeysm

src/pointwise_logdensities.jl

@@ -146,6 +146,16 @@ function _pointwise_tilde_observe(
     end
 end
 
+function tilde_assume!!(context::PointwiseLogdensityContext, right::Sampleable, vn, vi)
+    if _include_prior(context)
+        # rand_like!! accumulates the logp of the vi, so we don't need to manually
+        # track it.
+        value, vi = DynamicPPL.rand_like!!(right, context, vi)
+        return value, vi
+    else
+        return tilde_assume!!(context.context, right, vn, vi)
+    end
+end
 function tilde_assume!!(context::PointwiseLogdensityContext, right, vn, vi)
     !_include_prior(context) && return (tilde_assume!!(context.context, right, vn, vi))
     value, logp, vi = tilde_assume(context.context, right, vn, vi)