Skip to content

Commit 36e4651

Browse files
committed
Fix reconstruct calls in MH
1 parent aac4628 commit 36e4651

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

src/mcmc/mh.jl

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -285,10 +285,8 @@ end
285285
unvectorize(dists::AbstractVector) = length(dists) == 1 ? first(dists) : dists
286286

287287
# possibly unpack and reshape samples according to the prior distribution
288-
reconstruct(dist::Distribution, val::AbstractVector) = DynamicPPL.reconstruct(dist, val)
289-
function reconstruct(dist::AbstractVector{<:UnivariateDistribution}, val::AbstractVector)
290-
return val
291-
end
288+
reconstruct(dist::Distribution, val::AbstractVector) = DynamicPPL.from_vec_transform(dist)(val)
289+
reconstruct(dist::AbstractVector{<:UnivariateDistribution}, val::AbstractVector) = val
292290
function reconstruct(dist::AbstractVector{<:MultivariateDistribution}, val::AbstractVector)
293291
offset = 0
294292
return map(dist) do d
@@ -322,7 +320,7 @@ end
322320
:(
323321
$name = reconstruct(
324322
unvectorize(DynamicPPL.getdist.(Ref(vi), vns.$name)),
325-
DynamicPPL.getval(vi, vns.$name),
323+
DynamicPPL.getindex_internal(vi, vns.$name),
326324
)
327325
) for name in names
328326
]

0 commit comments

Comments
 (0)