use sample!!

Author: penelopeysm

src/extract_priors.jl

@@ -116,8 +116,7 @@ function extract_priors(rng::Random.AbstractRNG, model::Model)
     # workaround for the fact that `order` is still hardcoded in VarInfo, and hence you
     # can't push new variables without knowing the num_produce. Remove this when possible.
     varinfo = setaccs!!(varinfo, (PriorDistributionAccumulator(), NumProduceAccumulator()))
-    new_model = contextualize(model, SamplingContext(rng, model.context))
-    varinfo = last(evaluate!!(new_model, varinfo))
+    varinfo = last(sample!!(rng, model, varinfo))
     return getacc(varinfo, Val(:PriorDistributionAccumulator)).priors
 end