Throw an error when the time column does not exist

epiverse-trace · Sep 20, 2023 · a9f4b00 · a9f4b00
1 parent 872c6e1
commit a9f4b00
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/R/epichains.R b/R/epichains.R
@@ -290,6 +290,13 @@ aggregate.epichains <- function(x,
   grouping_var <- match.arg(grouping_var)
 
   out <- if (grouping_var == "time") {
+    if (is.null(x$time)) {
+      stop(
+        "Object must have a time column. ",
+        "To simulate time, specify `serials_dist` ",
+        "in the `simulate_tree()` setup."
+      )
+    }
     # Count the number of cases per generation
     stats::aggregate(
       list(cases = x$sim_id),