handle title separately

Author: palday

src/coefplot.jl

@@ -34,6 +34,7 @@ end
 """$(@doc coefplot)"""
 function coefplot!(fig::Indexable, x::Union{MixedModel,MixedModelBootstrap}; kwargs...)
     ax = Axis(fig[1, 1])
+    kwargs = _extract_title!(ax, kwargs)
     coefplot!(ax, x; kwargs...)
     return fig
 end
@@ -65,3 +66,12 @@ function coefplot!(ax::Axis, x::Union{MixedModel,MixedModelBootstrap};
     ylims!(ax, 0, nticks + 1)
     return ax
 end
+
+# function g(; kwargs...)
+#     if :a in keys(kwargs)
+#         a = kwargs[:a]
+#         kwargs = Base.pairs(NamedTuple((k => v for (k, v) in kwargs if k != :a)))
+#     end
+
+#     @info "" kwargs
+# end

src/ridge.jl

@@ -75,6 +75,7 @@ function ridgeplot!(ax::Axis, x::MixedModelBootstrap;
                   scatter_attributes, errorbars_attributes, attributes...)
     end
 
+    attributes = _extract_title!(ax, attributes)
     attributes = merge((; color=:black), attributes)
     band_attributes = merge((; color=(_color(attributes.color), 0.3)), band_attributes)
 

src/utilities.jl

@@ -55,6 +55,21 @@ function confint_table(x::MixedModelBootstrap, level=0.95; show_intercept=true)
     return filter!(:coefname => in(_coefnames(x; show_intercept)), df)
 end
 
+"""
+    _extract_title!(ax::Axis, kwargs)::Base.Pairs
+
+If a title is present in kwargs, use it to set the axis title.
+
+Returns kwargs without an entry for `title`.
+"""
+function _extract_title!(ax::Axis, kwargs)::Base.Pairs
+    if :title in keys(kwargs)
+        ax.title = kwargs[:title]
+        kwargs = NamedTuple((k => v for (k, v) in kwargs if k != :title))
+    end
+    return Base.pairs(kwargs)
+end
+
 _npreds(x; show_intercept=true) = length(_coefnames(x; show_intercept))
 
 """