Check that nbinom params are not both zero

epiforecasts · Aug 22, 2024 · 5c7f794 · 5c7f794
1 parent 37b36fd
commit 5c7f794
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/R/simulate.r b/R/simulate.r
@@ -128,6 +128,13 @@ chain_sim <- function(n, offspring, stat = c("size", "length"), infinite = Inf,
   if (!(exists(roffspring_name)) || !is.function(get(roffspring_name))) {
     stop("Function ", roffspring_name, " does not exist.")
   }
+  # If both parameters of the negative binomial are zero, you get NaNs
+  if (roffspring_name == "rnbinom" && all(c(...) == 0)) {
+    stop(
+      "The negative binomial parameters must have at least one ",
+      "non-zero parameter."
+    )
+  }
 
   if (!missing(serial)) {
     if (!is.function(serial)) {