Implement Turing.Inference.getlogp_external

Author: penelopeysm

Project.toml

@@ -1,6 +1,6 @@
 name = "SliceSampling"
 uuid = "43f4d3e8-9711-4a8c-bd1b-03ac73a255cf"
-version = "0.7.6"
+version = "0.7.7"
 
 [deps]
 AbstractMCMC = "80f14c24-f653-4e6a-9b94-39d6b0f70001"
@@ -21,7 +21,7 @@ Distributions = "0.25"
 LinearAlgebra = "1"
 LogDensityProblems = "2"
 Random = "1"
-Turing = "0.37, 0.38, 0.39"
+Turing = "0.39.5"
 julia = "1.10"
 
 [extras]

ext/SliceSamplingTuringExt.jl

@@ -39,13 +39,19 @@ function Turing.Inference.getparams(
 )
     return state.transition.params
 end
+
+function Turing.Inference.getlogp_external(
+    ::Turing.DynamicPPL.Model, t::SliceSampling.Transition, state
+)
+    return t.lp
+end
 # end
 
 function SliceSampling.initial_sample(rng::Random.AbstractRNG, ℓ::Turing.LogDensityFunction)
     model  = ℓ.model
     vi     = Turing.DynamicPPL.VarInfo(rng, model, Turing.SampleFromUniform())
     vi_spl = last(Turing.DynamicPPL.evaluate!!(model, rng, vi, Turing.SampleFromUniform()))
-    θ     = vi_spl[:]
+    θ      = vi_spl[:]
 
     init_attempt_count = 1
     while !all(isfinite.(θ))