implement initialstep for prior

TuringLang · Nov 10, 2023 · d90f49c · d90f49c
1 parent 26dcf83
commit d90f49c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/mcmc/Inference.jl b/src/mcmc/Inference.jl
@@ -114,6 +114,16 @@ DynamicPPL.unflatten(vi::SimpleVarInfo, θ::NamedTuple) = SimpleVarInfo(θ, vi.l
 # Algorithm for sampling from the prior
 struct Prior <: InferenceAlgorithm end
 
+function DynamicPPL.initialstep(
+    rng::AbstractRNG,
+    model::Model,
+    spl::Sampler{<:Prior},
+    vi::AbstractVarInfo;
+    kwargs...
+)
+    return Transition(model, vi), nothing
+end
+
 """
     mh_accept(logp_current::Real, logp_proposal::Real, log_proposal_ratio::Real)