Bug fix: checkbounds error in Pigeons integration

sefffal · Dec 9, 2024 · 7aae014 · 7aae014
1 parent c2eb654
commit 7aae014
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/ext/OctofitterPigeonsExt/OctofitterPigeonsExt.jl b/ext/OctofitterPigeonsExt/OctofitterPigeonsExt.jl
@@ -14,6 +14,11 @@ function Pigeons.initialization(model::Octofitter.LogDensityModel, rng::Abstract
 
     Octofitter.get_starting_point!!(rng, model)
 
+    if isnothing(model.starting_points) || !haskey(model.starting_points, chain_no)
+        @show model.starting_points chain_no
+        @show checkbounds(model.starting_points, chain_no)
+        error("Insufficient starting points provided. Provide at least one per chain. (model.starting_points is too short)")
+    end
     initial_θ_t = collect(model.starting_points[chain_no])
     # initial_θ_t = collect(Octofitter.get_starting_point!!(rng, model))
     initial_θ = model.invlink(initial_θ_t)