Skip to content

Commit 672a9b1

Browse files
authored
Merge pull request #105 from epiforecasts/hot-fix-missing-val-handling
Hot fix: edits the trajectories to quantiles function
2 parents 02cbb0f + 4807407 commit 672a9b1

File tree

1 file changed

+3
-10
lines changed

1 file changed

+3
-10
lines changed

R/utils.R

Lines changed: 3 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -85,20 +85,13 @@ trajectories_to_quantiles <- function(
8585
dplyr::across(tidyselect::all_of(c(timepoint_cols, id_cols)))
8686
)
8787

88-
missing_groups <- grouped_df |>
89-
dplyr::summarize(
90-
"any_missing" = anyNA(.data$value_col), # nolint
91-
.groups = "drop"
92-
) |>
93-
dplyr::filter(.data$any_missing) |>
94-
dplyr::select(-"any_missing")
95-
9688
quant_df <- grouped_df |>
97-
dplyr::anti_join(missing_groups, by = colnames(missing_groups)) |>
89+
filter(!is.na(value_col)) |>
9890
dplyr::reframe(
9991
!!quantile_value_name := stats::quantile(
10092
.data$value_col,
101-
probs = !!quantiles
93+
probs = !!quantiles,
94+
na.rm = TRUE
10295
),
10396
!!quantile_level_name := !!quantiles
10497
)

0 commit comments

Comments
 (0)