Skip to content

Commit c733b77

Browse files
committed
Fix tests
1 parent fd58050 commit c733b77

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

ext/SliceSamplingTuringExt.jl

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,12 @@ function Turing.Inference.getparams(
4040
return state.transition.params
4141
end
4242

43+
function Turing.Inference.getparams(
44+
::Turing.DynamicPPL.Model, state::SliceSampling.LatentSliceState
45+
)
46+
return state.transition.params
47+
end
48+
4349
function Turing.Inference.getlogp_external(
4450
::Turing.DynamicPPL.Model, t::SliceSampling.Transition, state
4551
)

test/turing.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@
3636
chain_logp_check = sample(
3737
logp_check(), externalsampler(sampler), 100; progress=false
3838
)
39-
@test isapprox(logpdf.(Normal(), chain_logp_check[:x]), chain_logp_check[:logp])
39+
@test isapprox(logpdf.(Normal(), chain_logp_check[:x]), chain_logp_check[:lp])
4040
end
4141

4242
@testset "gibbs($sampler)" for sampler in [
@@ -57,6 +57,6 @@
5757
chain_logp_check = sample(
5858
logp_check(), Turing.Gibbs(:x => externalsampler(sampler)), 100; progress=false
5959
)
60-
@test isapprox(logpdf.(Normal(), chain_logp_check[:x]), chain_logp_check[:logp])
60+
@test isapprox(logpdf.(Normal(), chain_logp_check[:x]), chain_logp_check[:lp])
6161
end
6262
end

0 commit comments

Comments
 (0)