Added interval_to_period helper

mitchelloharawild · Jan 23, 2020 · 98fc85a · 98fc85a
1 parent 4e33934
commit 98fc85a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/R/model.R b/R/model.R
@@ -84,11 +84,7 @@ specials_prophet <- new_specials(
                     name = NULL){
     # Extract data interval
     interval <- tsibble::interval(self$data)
-    interval <- with(interval, lubridate::years(year) +
-                       lubridate::period(3*quarter + month, units = "month") + lubridate::weeks(week) +
-                       lubridate::days(day) + lubridate::hours(hour) + lubridate::minutes(minute) +
-                       lubridate::seconds(second) + lubridate::milliseconds(millisecond) +
-                       lubridate::microseconds(microsecond) + lubridate::nanoseconds(nanosecond))
+    interval <- interval_to_period(interval)
 
     if(is.null(name) & is.character(period)){
       name <- period

diff --git a/R/utils.R b/R/utils.R
@@ -0,0 +1,7 @@
+interval_to_period <- function(interval){
+  with(interval, lubridate::years(year) +
+         lubridate::period(3*quarter + month, units = "month") + lubridate::weeks(week) +
+         lubridate::days(day) + lubridate::hours(hour) + lubridate::minutes(minute) +
+         lubridate::seconds(second) + lubridate::milliseconds(millisecond) +
+         lubridate::microseconds(microsecond) + lubridate::nanoseconds(nanosecond))
+}