bug fix with missing data in drivers with future values

microsoft · Oct 22, 2024 · 4a12ad6 · 4a12ad6
1 parent e7080fc
commit 4a12ad6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/R/hierarchy.R b/R/hierarchy.R
@@ -112,6 +112,7 @@ prep_hierarchical_data <- function(input_data,
           .noexport = NULL
         ) %do% {
           temp_tbl <- input_data_adj %>%
+            tidyr::drop_na(tidyselect::all_of(regressor_var)) %>%
             dplyr::select(Date, tidyselect::all_of(value_level_iter), tidyselect::all_of(regressor_var)) %>%
             dplyr::distinct()
 
@@ -144,6 +145,7 @@ prep_hierarchical_data <- function(input_data,
 
         # agg by total
         total_tbl <- input_data_adj %>%
+          tidyr::drop_na(tidyselect::all_of(regressor_var)) %>%
           dplyr::select(Date, value_level[[1]], tidyselect::all_of(regressor_var)) %>%
           dplyr::distinct() %>%
           dplyr::group_by(Date) %>%