plot.brmshypothesis: don't show legend unless necessary

paul-buerkner · Dec 18, 2023 · 6377885 · 6377885
1 parent e496ca1
commit 6377885
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/R/hypothesis.R b/R/hypothesis.R
@@ -538,6 +538,9 @@ plot.brmshypothesis <- function(x, N = 5, ignore_prior = FALSE,
   }
 
   .plot_fun <- function(samples) {
+    samples <- na.omit(samples)
+    # if no prior draws are present, there is no need to plot a legend
+    ignore_prior <- ignore_prior || length(unique(samples$Type)) == 1L
     gg <- ggplot(samples, aes(x = .data[["values"]])) +
       facet_wrap("ind", ncol = 1, scales = "free") +
       xlab("") + ylab("") + theme +