Merge pull request #259 from SciML/dw/dynamichmc

Account for changes in DynamicHMC 3.3.0
SciML · Sep 6, 2022 · 06b6c31 · 06b6c31 · devmotion · Sep 6, 2022
2 parents dfba687 + 7c7169e
commit 06b6c31
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "DiffEqBayes"
 uuid = "ebbdde9d-f333-5424-9be2-dbf1e9acfb5e"
 authors = ["Vaibhavdixit02 <[email protected]>"]
-version = "3.0.0"
+version = "3.0.1"
 
 [deps]
 DiffEqBase = "2b5f629d-d688-5b77-993f-72d75c75574e"

diff --git a/src/dynamichmc_inference.jl b/src/dynamichmc_inference.jl
@@ -114,5 +114,11 @@ function dynamichmc_inference(problem::DiffEqBase.DEProblem, algorithm, t, data,
     ℓ = TransformedLogDensity(trans, P)
     ∇ℓ = LogDensityProblems.ADgradient(AD_gradient_kind, ℓ)
     results = mcmc_with_warmup(rng, ∇ℓ, num_samples; mcmc_kwargs...)
-    merge((posterior = TransformVariables.transform.(Ref(trans), results.chain),), results)
+    chain = if haskey(results, :chain) # DynamicHMC < 3.3.0
+        results.chain
+    else
+        eachcol(results.posterior_matrix)
+    end
+    posterior = map(Base.Fix1(TransformVariables.transform, trans), chain)
+    merge((; posterior), results)
 end