Fix regression in octocorner

sefffal · Mar 7, 2024 · 60594e8 · 60594e8
1 parent f3c21b3
commit 60594e8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ext/OctofitterPairPlotsExt.jl b/ext/OctofitterPairPlotsExt.jl
@@ -7,9 +7,14 @@ using MCMCChains
 
 unitlengthprior_vars(::Octofitter.UnitLengthPrior{VX,VY}) where {VX, VY} = (VX, VY)
 
+Octofitter.octocorner(
+    model::Octofitter.LogDensityModel,
+    chains::MCMCChains.Chains...; 
+    kwargs...
+) = octocorner(model.system, chains...; kwargs...)
 function Octofitter.octocorner(
     system::System,
-    chains...; 
+    chains::MCMCChains.Chains...; 
     small=false,
     labels=Dict{Symbol,Any}(),
     fname=small ? "$(system.name)-pairplot-small.png" : "$(system.name)-pairplot.png" ,