Bug Fix: support non-stabilized variational reference

sefffal · Oct 21, 2024 · 49a77c1 · 49a77c1
1 parent 78420cc
commit 49a77c1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ext/OctofitterPigeonsExt/OctofitterPigeonsExt.jl b/ext/OctofitterPigeonsExt/OctofitterPigeonsExt.jl
@@ -158,7 +158,12 @@ Base.@nospecializeinfer function MCMCChains.Chains(
 
     # Resolve the array back into the nested named tuple structure used internally.
     # Augment with some internal fields
-    samples = get_sample(pt, chain_num)
+    local samples
+    try
+        samples = get_sample(pt, chain_num)
+    catch
+        samples = get_sample(pt)
+    end
     chain_res = map(samples) do sample 
         θ_t = @view(sample[begin:begin+model.D-1])
         logpot = sample[model.D+1]