update Turing and AHMC compat

mohamed82008 · mohamed82008 · commit 1b6e5fc6383c · 2020-01-16T22:32:00.000+11:00
diff --git a/Project.toml b/Project.toml
@@ -11,7 +11,7 @@ MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
 
 [compat]
 AbstractMCMC = "~0.1"
-AdvancedHMC = "0.2.15"
+AdvancedHMC = "0.2.20"
 Bijectors = "0.5.2"
 Distributions = "0.22"
 DistributionsAD = "0.2"
diff --git a/test/Turing/core/compat/zygote.jl b/test/Turing/core/compat/zygote.jl
@@ -11,7 +11,6 @@ function gradient_logp_reverse(
     model::Model,
     sampler::AbstractSampler = SampleFromPrior(),
 )
-    T = typeof(getlogp(vi))
 
     # Specify objective function.
     function f(θ)
diff --git a/test/Turing/inference/hmc.jl b/test/Turing/inference/hmc.jl
@@ -585,7 +585,7 @@ function callback(
     cb::HMCCallback;
     kwargs...
 )
-    AHMC.pm_next!(cb.p, t.stat, iteration, spl.state.h.metric)
+    AHMC.pm_next!(cb.p, (iteration=iteration, t.stat..., mass_matrix=spl.state.h.metric))
 end
 
 function init_callback(

Original file line number	Diff line number	Diff line change
`@@ -11,7 +11,6 @@ function gradient_logp_reverse(`
`11`	`11`	`model::Model,`
`12`	`12`	`sampler::AbstractSampler = SampleFromPrior(),`
`13`	`13`	`)`
`14`		`- T = typeof(getlogp(vi))`
`15`	`14`
`16`	`15`	`# Specify objective function.`
`17`	`16`	`function f(θ)`
Original file line number	Diff line number	Diff line change
`@@ -585,7 +585,7 @@ function callback(`
`585`	`585`	`cb::HMCCallback;`
`586`	`586`	`kwargs...`
`587`	`587`	`)`
`588`		`- AHMC.pm_next!(cb.p, t.stat, iteration, spl.state.h.metric)`
	`588`	`+ AHMC.pm_next!(cb.p, (iteration=iteration, t.stat..., mass_matrix=spl.state.h.metric))`
`589`	`589`	`end`
`590`	`590`
`591`	`591`	`function init_callback(`