fix: revision analysis?

cmu-delphi · Sep 25, 2024 · b3f772c · b3f772c
1 parent f481bd7
commit b3f772c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/revision_analysis.R b/R/revision_analysis.R
@@ -81,8 +81,8 @@ revision_summary <- function(epi_arch,
                              should_compactify = TRUE) {
   arg <- names(eval_select(rlang::expr(c(...)), allow_rename = FALSE, data = epi_arch$DT))
   if (length(arg) == 0) {
-    first_non_key <- !(names(epi_arch$DT) %in% c(key_colnames(epi_arch), "version"))
-    arg <- names(epi_arch$DT)[first_non_key][1]
+    # Choose the first column that's not a key or version
+    arg <- setdiff(names(epi_arch$DT), c(key_colnames(epi_arch), "version"))[[1]]
   } else if (length(arg) > 1) {
     cli_abort("Not currently implementing more than one column at a time. Run each separately")
   }