Fix moving window default

sarahhbellum · Jun 10, 2019 · 76551d7 · 76551d7
1 parent e15f885
commit 76551d7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/R/NobBS.R b/R/NobBS.R
@@ -39,6 +39,7 @@ NobBS <- function(data, now, units, onset_date, report_date, moving_window=NULL,
                     nBurnin=1000,
                     nThin=1,
                     nSamp=10000)) {
+
   # Check that "now" is entered as a Date
   if(inherits(now, "Date")==FALSE){
     stop("'Now' argument must be of datatype Date (as.Date)")
@@ -50,6 +51,9 @@ NobBS <- function(data, now, units, onset_date, report_date, moving_window=NULL,
   now.T <- length(seq(min(data[,onset_date]),as.Date(now),by=units))
 
   # Check the default arguments
+  if (is.null(moving_window)) {
+    moving_window <- now.T
+  }
   if (is.null(max_D)) {
     max_D <- now.T-1
   }
@@ -247,4 +251,4 @@ NobBS <- function(data, now, units, onset_date, report_date, moving_window=NULL,
 
   nowcast_results <<- list(estimates=estimates, nowcast.post.samps=nowcast.post.samps,params.post=parameter_extract[2:ncol(parameter_extract)])
 
-}
+}