Refactor seqB

[Rnaught] / R / seq_bayes.R

diff --git a/R/seq_bayes.R b/R/seq_bayes.R
index ccc9a415195b808da8995c7b39043c01a18e2269..b25d273276930293c083a988901db7517054c781 100644 (file)
--- a/R/seq_bayes.R
+++ b/R/seq_bayes.R
@@ -103,17 +103,13 @@ seq_bayes <- function(cases, mu, kappa = 20, post = FALSE) {
     )
   }
 
     )
   }
 

-  if (any(cases == 0)) {
-    times <- which(cases > 0)
-    if (length(times) < 2) {
-      stop("Case counts must contain at least two positive integers.",
-        call. = FALSE
-      )
-    }
-    cases <- cases[times]
-  } else {
-    times <- seq_along(cases)
+  times <- which(cases > 0)
+  if (length(times) < 2) {
+    stop("Case counts must contain at least two positive integers.",
+      call. = FALSE
+    )

   }
   }

+  cases <- cases[times]

 
   support <- seq(0, kappa, 0.01)
   tau <- diff(times)
 
   support <- seq(0, kappa, 0.01)
   tau <- diff(times)