Skip to content

Commit 676e06f

Browse files
authored
Merge pull request #139 from TuringLang/mcmcchains
Fixes for MCMCChains 4.0.0
2 parents 700ae06 + ba5e3eb commit 676e06f

File tree

17 files changed

+999
-86
lines changed

17 files changed

+999
-86
lines changed

Project.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
1212
ZygoteRules = "700de1a5-db45-46bc-99cf-38207098b444"
1313

1414
[compat]
15-
AbstractMCMC = "1.0"
16-
Bijectors = "0.5.2, 0.6, 0.7"
15+
AbstractMCMC = "1"
16+
Bijectors = "0.5.2, 0.6, 0.7, 0.8"
1717
Distributions = "0.22, 0.23"
1818
MacroTools = "0.5.1"
1919
ZygoteRules = "0.2"

src/prob_macro.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,7 @@ _setval!(vi::TypedVarInfo, c::AbstractChains) = _setval!(vi.metadata, vi, c)
233233
return Expr(:block, map(names) do n
234234
quote
235235
for vn in md.$n.vns
236-
val = copy.(vec(c[Symbol(string(vn))].value))
236+
val = vec(c[Symbol(vn)])
237237
setval!(vi, val, vn)
238238
settrans!(vi, false, vn)
239239
end

test/Turing/Turing.jl

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ using Requires, Reexport, ForwardDiff
1212
using DistributionsAD, Bijectors, StatsFuns, SpecialFunctions
1313
using Statistics, LinearAlgebra
1414
using Libtask
15-
@reexport using Distributions, MCMCChains, Libtask, AbstractMCMC
15+
@reexport using Distributions, MCMCChains, Libtask, AbstractMCMC, Bijectors
1616
using Tracker: Tracker
1717

1818
import DynamicPPL: getspace, NoDist, NamedDist
@@ -58,6 +58,11 @@ using .Variational
5858
end
5959
end
6060

61+
@init @require Optim="429524aa-4258-5aef-a3af-852621145aeb" @eval begin
62+
include("modes/ModeEstimation.jl")
63+
export MAP, MLE, optimize
64+
end
65+
6166
###########
6267
# Exports #
6368
###########
@@ -71,6 +76,7 @@ export @model, # modelling
7176

7277
MH, # classic sampling
7378
RWMH,
79+
Emcee,
7480
ESS,
7581
Gibbs,
7682

@@ -87,7 +93,7 @@ export @model, # modelling
8793
CSMC,
8894
PG,
8995

90-
vi, # variational inference
96+
vi, # variational inference
9197
ADVI,
9298

9399
sample, # inference

test/Turing/inference/AdvancedSMC.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,8 @@ struct PG{space,R} <: ParticleInference
170170
resampler::R
171171
end
172172

173+
isgibbscomponent(::PG) = true
174+
173175
"""
174176
PG(n, space...)
175177
PG(n, [resampler = ResampleWithESSThreshold(), space = ()])

0 commit comments

Comments
 (0)