fix printing

easystats · Sep 3, 2024 · b28a7f3 · b28a7f3
1 parent 57be220
commit b28a7f3
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/R/bayesfactor_restricted.R b/R/bayesfactor_restricted.R
@@ -176,11 +176,10 @@ bayesfactor_restricted.emmGrid <- function(posterior, hypothesis, prior = NULL,
     verbose = verbose
   )
 
-  out <- bayesfactor_restricted.data.frame(
+  bayesfactor_restricted.data.frame(
     posterior = samps$posterior, prior = samps$prior,
     hypothesis = hypothesis
   )
-  .append_datagrid(out, posterior)
 }
 
 #' @export

diff --git a/R/format.R b/R/format.R
@@ -225,7 +225,8 @@ format.bayesfactor_restricted <- function(x,
     BFE$BF[sgn] <- paste0("-", BFE$BF[sgn])
   }
   BFE$log_BF <- NULL
-  colnames(BFE) <- c("Hypothesis", "P(Prior)", "P(Posterior)", "BF")
+  colnames(BFE)[colnames(BFE)=="p_prior"] <- "P(Prior)"
+  colnames(BFE)[colnames(BFE)=="p_posterior"] <- "P(Posterior)"
 
   # footer
   if (is.null(format) || format == "text") {

diff --git a/R/print.equivalence_test.R b/R/print.equivalence_test.R
@@ -34,7 +34,8 @@ print.equivalence_test <- function(x, digits = 2, ...) {
   x$HDI <- sprintf("[%*s %*s]", maxlen_low, x$HDI_low, maxlen_high, x$HDI_high)
 
   ci <- unique(x$CI)
-  keep.columns <- c("CI", "Parameter", "ROPE_Equivalence", "ROPE_Percentage", "HDI", "Effects", "Component")
+  keep.columns <- c(attr(x, "grid_cols"), "Parameter", "Effects", "Component",
+                    "ROPE_Equivalence", "ROPE_Percentage", "CI", "HDI")
 
   x <- x[, intersect(keep.columns, colnames(x))]
 

diff --git a/R/print.rope.R b/R/print.rope.R
@@ -28,7 +28,7 @@ print.rope <- function(x, digits = 2, ...) {
 
   # These are the base columns we want to print
   cols <- c(
-    "Parameter", "ROPE_Percentage", "Effects", "Component",
+    attr(x, "grid_cols"), "Parameter", "ROPE_Percentage", "Effects", "Component",
     if (is_multivariate) c("ROPE_low", "ROPE_high")
   )