Skip to content

Commit bf73fd0

Browse files
authored
fix an error and reorder some functions (#650)
1 parent 3ffb003 commit bf73fd0

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

ext/DynamicPPLMCMCChainsExt.jl

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,6 @@ else
88
using ..MCMCChains: MCMCChains
99
end
1010

11-
_has_varname_to_symbol(info::NamedTuple{names}) where {names} = :varname_to_symbol in names
12-
function _check_varname_indexing(c::MCMCChains.Chains)
13-
return DynamicPPL.supports_varname_indexing(c) ||
14-
error("Chains do not support indexing using $vn.")
15-
end
16-
1711
# Load state from a `Chains`: By convention, it is stored in `:samplerstate` metadata
1812
function DynamicPPL.loadstate(chain::MCMCChains.Chains)
1913
if !haskey(chain.info, :samplerstate)
@@ -26,10 +20,17 @@ function DynamicPPL.loadstate(chain::MCMCChains.Chains)
2620
return chain.info[:samplerstate]
2721
end
2822

29-
# A few methods needed.
23+
_has_varname_to_symbol(info::NamedTuple{names}) where {names} = :varname_to_symbol in names
24+
3025
function DynamicPPL.supports_varname_indexing(chain::MCMCChains.Chains)
3126
return _has_varname_to_symbol(chain.info)
3227
end
28+
29+
function _check_varname_indexing(c::MCMCChains.Chains)
30+
return DynamicPPL.supports_varname_indexing(c) ||
31+
error("Chains do not support indexing using `VarName`s.")
32+
end
33+
3334
function DynamicPPL.getindex_varname(
3435
c::MCMCChains.Chains, sample_idx, vn::DynamicPPL.VarName, chain_idx
3536
)

0 commit comments

Comments
 (0)