Filter prior density layer to include only shown parameters.

easystats · Jul 7, 2023 · 3c7ad13 · 3c7ad13
1 parent 24436e9
commit 3c7ad13
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/R/plot.estimate_density.R b/R/plot.estimate_density.R
@@ -131,6 +131,9 @@ plot.see_estimate_density <- function(x,
     n_columns <- NULL
   }
 
+  # get parameter names for filtering
+  params <- unique(x$y)
+
   # get labels
   labels <- .clean_parameter_names(x$Parameter, grid = !is.null(n_columns))
 
@@ -152,6 +155,7 @@ plot.see_estimate_density <- function(x,
       p <- p +
         .add_prior_layer_ridgeline(
           model,
+          parameter = params,
           show_intercept = show_intercept,
           priors_alpha = priors_alpha,
           show_ridge_line = FALSE

diff --git a/R/plot.p_direction.R b/R/plot.p_direction.R
@@ -180,6 +180,9 @@ plot.see_p_direction <- function(x,
     n_columns <- NULL
   }
 
+  # get parameter names for filtering
+  params <- unique(x$y)
+
   # get labels
   labels <- .clean_parameter_names(x$y, grid = !is.null(n_columns))
 
@@ -203,6 +206,7 @@ plot.see_p_direction <- function(x,
   if (priors) {
     p <- p + .add_prior_layer_ridgeline(
       model,
+      parameter = params,
       show_intercept = show_intercept,
       priors_alpha = priors_alpha
     )

diff --git a/R/plot.p_significance.R b/R/plot.p_significance.R
@@ -192,6 +192,9 @@ plot.see_p_significance <- function(x,
     n_columns <- NULL
   }
 
+  # get parameter names for filtering
+  params <- unique(x$y)
+
   # get labels
   labels <- .clean_parameter_names(x$y, grid = !is.null(n_columns))
 
@@ -216,6 +219,7 @@ plot.see_p_significance <- function(x,
     p <- p +
       .add_prior_layer_ridgeline(
         model,
+        parameter = params,
         show_intercept = show_intercept,
         priors_alpha = priors_alpha
       ) +