Merge pull request #466 from mitchelloharawild/ggplot_geomForecast2

Fixed issue with geom_forecast when using ts object and series argument
robjhyndman · Jan 10, 2017 · b6b1605 · b6b1605
2 parents 9e02207 + e475a48
commit b6b1605
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/R/ggplot.R b/R/ggplot.R
@@ -1535,7 +1535,12 @@ geom_forecast <- function(mapping = NULL, data = NULL, stat = "forecast",
   if(stat=="forecast"){
     paramlist <- list(na.rm = na.rm, PI=PI, series=series, ...)
     if(!is.null(series)){
-      mapping <- ggplot2::aes_(colour = ~..series..)
+      if(inherits(mapping, "uneval")){
+        mapping$colour = quote(..series..)
+      }
+      else{
+        mapping <- ggplot2::aes_(colour = ~..series..)
+      }
     }
   }
   else{