print: column_names

vincentarelbundock · Oct 10, 2023 · 8c472b9 · 8c472b9
1 parent d443bb6
commit 8c472b9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -1,6 +1,6 @@
 Package: marginaleffects
 Title: Predictions, Comparisons, Slopes, Marginal Means, and Hypothesis Tests
-Version: 0.15.1.9012
+Version: 0.15.1.9013
 Authors@R: 
     c(person(given = "Vincent",
              family = "Arel-Bundock",

diff --git a/NEWS.md b/NEWS.md
@@ -11,6 +11,7 @@ Misc:
 
 * Add support for `hypotheses()` to `inferences()`. This changes `FUN` argument name in `hypotheses` to `FUN` to not conflict with `inferences`. Thanks to @Tristan-Siegfried for code contribution #908.
 * Support `survival::survreg()`. Thanks to Carlisle Rainey for Report #911.
+* `column_names` argument in `print.marginaleffects()` to suppress the printed column names at the bottom of the printout.
 
 ## 0.15.1
 

diff --git a/R/print.R b/R/print.R
@@ -37,6 +37,7 @@ print.marginaleffects <- function(x,
                                   ncols = getOption("marginaleffects_print_ncols", default = 30),
                                   style = getOption("marginaleffects_print_style", default = "summary"),
                                   type = getOption("marginaleffects_print_type", default = TRUE),
+                                  column_names = getOption("marginaleffects_print_column_names", default = TRUE),
                                   ...) {
 
 
@@ -242,7 +243,7 @@ print.marginaleffects <- function(x,
         cat(sprintf("Results averaged over levels of: %s",
                     paste(vg, collapse = ", ")), "\n")
     }
-    if (ncol(x) <= ncols) {
+    if (ncol(x) <= ncols && isTRUE(column_names)) {
         cat("Columns:", paste(colnames(x), collapse = ", "), "\n")
     }
     if (isTRUE(type) && !is.null(attr(x, "type"))) {