Parcourir la source

invisibly return component plots (#11)

This will make it easier for users to change the theme or other aspects of the plots, and shouldn't affect any existing behavior because the returned object is invisible.
David J. Harris il y a 9 ans
Parent
commit
3548546b65
1 fichiers modifiés avec 3 ajouts et 0 suppressions
  1. 3 0
      R/R/prophet.R

+ 3 - 0
R/R/prophet.R

@@ -919,6 +919,8 @@ plot.prophet <- function(x, fcst, uncertainty = TRUE, xlabel = 'ds',
 #' @param uncertainty Boolean indicating if the uncertainty interval should be
 #'  plotted for the trend, from fcst columns trend_lower and trend_upper.
 #'
+#' @return Invisibly return a list containing the plotted ggplot objects
+#'
 #' @export
 #' @importFrom dplyr "%>%"
 prophet_plot_components <- function(m, fcst, uncertainty = TRUE) {
@@ -945,6 +947,7 @@ prophet_plot_components <- function(m, fcst, uncertainty = TRUE) {
     print(panels[[i]], vp = grid::viewport(layout.pos.row = i,
                                            layout.pos.col = 1))
   }
+  return(invisible(panels))
 }
 
 #' Plot the prophet trend.