use dplyr group_vars

RichardMN · Feb 5, 2022 · ee41f59 · ee41f59
1 parent 412d544
commit ee41f59
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/NAMESPACE b/NAMESPACE
@@ -61,6 +61,7 @@ importFrom(dplyr,filter)
 importFrom(dplyr,full_join)
 importFrom(dplyr,group_by)
 importFrom(dplyr,group_by_at)
+importFrom(dplyr,group_vars)
 importFrom(dplyr,if_else)
 importFrom(dplyr,lag)
 importFrom(dplyr,last_col)

diff --git a/R/processing.R b/R/processing.R
@@ -51,15 +51,13 @@ set_negative_values_to_zero <- function(data) {
 #' @family compulsory_processing
 #' @concept compulsory_processing
 #' @importFrom tidyr complete full_seq nesting
-#' @importFrom dplyr starts_with group_by ungroup
+#' @importFrom dplyr starts_with group_by ungroup group_vars
 #' @importFrom rlang !!! syms
 fill_empty_dates_with_na <- function(data) {
   regions <- select(data, starts_with("level_")) %>%
     names()
 
-  groups <- attributes(data)$groups
-  groups <- colnames(groups)
-  groups <- setdiff(groups, ".rows")
+  groups <- group_vars(data)
 
   data <- data %>%
     ungroup() %>%