Fixed issue with autoplot() and length 1 forecasts

Resolves #400
tidyverts · May 13, 2024 · 6865b67 · 6865b67
1 parent 9918b60
commit 6865b67
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/NEWS.md b/NEWS.md
@@ -1,5 +1,9 @@
 # fabletools (development version)
 
+## Bug fixes
+
+* Fixed issue with `autoplot()` and length 1 forecasts (#400).
+
 # fabletools 0.4.2
 
 Minor patch to build package with latest R version as requested by CRAN.

diff --git a/R/plot.R b/R/plot.R
@@ -349,15 +349,15 @@ build_fbl_layer <- function(object, data = NULL, level = c(80, 95),
         out[[length(out) + 1L]] <- ggdist::geom_lineribbon(without(intvl_mapping, "colour_ramp"), data = dist_qi_frame(object[single_row[["FALSE"]],], level), ..., inherit.aes = FALSE, key_glyph = draw_key_ribbon)
       }
       if(length(single_row[["TRUE"]]) > 0) {
-        out[[length(out) + 1L]] <- ggdist::stat_interval(intvl_mapping, data = dist_qi_frame(object[single_row[["TRUE"]],], level), ..., inherit.aes = FALSE, key_glyph = draw_key_ribbon)
+        out[[length(out) + 1L]] <- ggdist::geom_interval(intvl_mapping, data = dist_qi_frame(object[single_row[["TRUE"]],], level), orientation = "vertical", ..., inherit.aes = FALSE, key_glyph = draw_key_ribbon)
       }
       out[[length(out) + 1L]] <- ggplot2::labs(fill = col_nm)
     } else {
       if(length(single_row[["FALSE"]]) > 0) {
         out[[length(out) + 1L]] <- ggdist::geom_lineribbon(without(intvl_mapping, "colour_ramp"), data = dist_qi_frame(object[single_row[["FALSE"]],], level), fill = colour, ..., inherit.aes = FALSE, key_glyph = draw_key_ribbon)
       }
       if(length(single_row[["TRUE"]]) > 0) {
-        out[[length(out) + 1L]] <- ggdist::stat_interval(intvl_mapping, data = dist_qi_frame(object[single_row[["TRUE"]],], level), colour = colour, ..., inherit.aes = FALSE, key_glyph = draw_key_ribbon)
+        out[[length(out) + 1L]] <- ggdist::geom_interval(intvl_mapping, data = dist_qi_frame(object[single_row[["TRUE"]],], level), colour = colour, orientation = "vertical", ..., inherit.aes = FALSE, key_glyph = draw_key_ribbon)
       }
     }