Move input checking on nsim_obs where it is first needed

epiverse-trace · Mar 22, 2024 · 80115cd · 80115cd
1 parent 6f3e3cd
commit 80115cd
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/R/likelihood.R b/R/likelihood.R
@@ -95,12 +95,6 @@ likelihood <- function(chains, statistic = c("size", "length"), offspring_dist,
   statistic <- match.arg(statistic)
 
   ## Input checking
-  ## Check nsim_obs when specified
-  if (!missing(nsim_obs)) {
-    checkmate::assert_number(
-      nsim_obs, lower = 1, finite = TRUE, na.ok = FALSE
-    )
-  }
   checkmate::assert(
     checkmate::check_numeric(
       chains, lower = 0, upper = Inf, any.missing = FALSE
@@ -145,6 +139,10 @@ likelihood <- function(chains, statistic = c("size", "length"), offspring_dist,
   if (obs_prob < 1) {
     if (missing(nsim_obs)) {
       stop("'nsim_obs' must be specified if 'obs_prob' is < 1")
+    } else {
+      checkmate::assert_number(
+        nsim_obs, lower = 1, finite = TRUE, na.ok = FALSE
+      )
     }
 
     statistic_func <- .get_statistic_func(statistic)