Add support for Pigeons.sample_iid!

sefffal · Feb 2, 2024 · 7a42345 · 7a42345 · sefffal · Feb 2, 2024
1 parent a9e79fe
commit 7a42345
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/docs/src/pma.md b/docs/src/pma.md
@@ -211,12 +211,6 @@ pairplot(
             PairPlots.MarginConfidenceLimits()
         ),
     labels=Dict(:mass=>"mass [Mⱼᵤₚ]", :a=>"sma. [au]"),
-    axis = (;
-        a = (;
-            scale=Makie.pseudolog10,
-            ticks=2 .^ (0:1:6)
-        )
-    )
 )
 ```
 

diff --git a/ext/OctofitterPigeonsExt.jl b/ext/OctofitterPigeonsExt.jl
@@ -13,6 +13,15 @@ function Pigeons.initialization(model::Octofitter.LogDensityModel, rng::Abstract
     return initial_θ_t
 end
 
+# Valid for reference model only
+function Pigeons.sample_iid!(model_reference::Octofitter.LogDensityModel, replica, shared)
+    # This could in theory be done without any array allocations
+    θ = sample_priors(replica.rng, model_reference.system)
+    θ_t = model_reference.link(θ)
+    replica.state .= θ_t
+end
+
+
 
 """
 octofit_pigeons(model; nrounds, n_chains=[auto])