move division outside vapply in proportion_cluster_size

Co-authored-by: Hugo Gruson <Bisaloo@users.noreply.github.com>
epiverse-trace · Jan 22, 2024 · 9e7a07a · 9e7a07a
1 parent 1ab80a5
commit 9e7a07a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/proportion_cluster_size.R b/R/proportion_cluster_size.R
@@ -70,8 +70,8 @@ proportion_cluster_size <- function(R, k, cluster_size, ..., offspring_dist,
       size = df[i, "k"]
     )
     propn_cluster <- vapply(cluster_size, function(x) {
-      sum(simulate_secondary[simulate_secondary >= x]) / sum(simulate_secondary)
-    }, FUN.VALUE = numeric(1))
+      sum(simulate_secondary[simulate_secondary >= x])
+    }, FUN.VALUE = numeric(1)) / sum(simulate_secondary)
     if (format_prop) {
       propn_cluster <- paste0(round(propn_cluster * 100, digits = 1), "%")
     }