Improve calculation of lengths

epiverse-trace · Feb 20, 2024 · 70a908a · 70a908a
1 parent 2732d67
commit 70a908a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/epichains.R b/R/epichains.R
@@ -348,8 +348,8 @@ summary.epichains_tree <- function(object, ...) {
   } else {
     # Length: each case's last sim_id
     for (i in seq_len(index_cases)) {
-      max_case_index <- max(which(object$infectee_id == i))
-      chain_summaries[i] <- object$sim_id[max_case_index]
+      index_case_rows <- which(object$infector_id == i)
+      chain_summaries[i] <- length(unique((object$sim_id[index_case_rows])))
     }
   }