Abbreviate code block with ifelse + vapply construct

Co-authored-by: Sebastian Funk <[email protected]>
epiverse-trace · Sep 12, 2023 · 5ab7c85 · 5ab7c85
1 parent 0b2005a
commit 5ab7c85
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/R/likelihood.R b/R/likelihood.R
@@ -143,11 +143,8 @@ likelihood <- function(chains, statistic = c("size", "length"), offspring_dist,
   ## (sum of the log-likelihoods), if log == TRUE, else
   ## multiply the likelihoods
   if (isFALSE(individual)) {
-    if (isTRUE(log)) {
-      chains_likelihood <- vapply(chains_likelihood, sum, 0)
-    } else {
-      chains_likelihood <- vapply(chains_likelihood, prod, 0)
-    }
+    summarise_func <- ifelse(log, sum, prod)
+    vapply(chains_likelihood, summarise_func, 0)
   }
 
   return(chains_likelihood)