Skip to content

Commit 22a0360

Browse files
committed
Fix fix fix tests, reduce code duplication
1 parent 398bd0c commit 22a0360

File tree

1 file changed

+8
-19
lines changed

1 file changed

+8
-19
lines changed

ext/SliceSamplingTuringExt.jl

Lines changed: 8 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -22,30 +22,19 @@ Turing.Inference.isgibbscomponent(::SliceSampling.Slice) = true
2222
Turing.Inference.isgibbscomponent(::SliceSampling.SliceSteppingOut) = true
2323
Turing.Inference.isgibbscomponent(::SliceSampling.SliceDoublingOut) = true
2424

25+
const SliceSamplingStates = Union{
26+
SliceSampling.UnivariateSliceState,
27+
SliceSampling.GibbsState,
28+
SliceSampling.HitAndRunState,
29+
SliceSampling.LatentSliceState,
30+
SliceSampling.GibbsPolarSliceState,
31+
}
2532
function Turing.Inference.getparams(
26-
::Turing.DynamicPPL.Model, sample::SliceSampling.UnivariateSliceState
33+
::Turing.DynamicPPL.Model, sample::SliceSamplingStates
2734
)
2835
return sample.transition.params
2936
end
3037

31-
function Turing.Inference.getparams(
32-
::Turing.DynamicPPL.Model, state::SliceSampling.GibbsState
33-
)
34-
return state.transition.params
35-
end
36-
37-
function Turing.Inference.getparams(
38-
::Turing.DynamicPPL.Model, state::SliceSampling.HitAndRunState
39-
)
40-
return state.transition.params
41-
end
42-
43-
function Turing.Inference.getparams(
44-
::Turing.DynamicPPL.Model, state::SliceSampling.LatentSliceState
45-
)
46-
return state.transition.params
47-
end
48-
4938
function Turing.Inference.getlogp_external(
5039
::Turing.DynamicPPL.Model, t::SliceSampling.Transition, state
5140
)

0 commit comments

Comments
 (0)