Handle outcome NAs in get_draws_sigma()

openpharma · Jun 13, 2024 · aa6448d · aa6448d
1 parent 33c9a77
commit aa6448d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -1,6 +1,6 @@
 Package: brms.mmrm
 Title: Bayesian MMRMs using 'brms'
-Version: 1.0.0.9000
+Version: 1.0.0.9001
 Authors@R: c(
   person(
     given = c("William", "Michael"),

diff --git a/NEWS.md b/NEWS.md
@@ -1,6 +1,6 @@
-# brms.mmrm 1.0.0.9000 (development)
-
+# brms.mmrm 1.0.0.9001 (development)
 
+* Handle outcome `NA`s in `get_draws_sigma()`.
 
 # brms.mmrm 1.0.0
 

diff --git a/R/brm_marginal_draws.R b/R/brm_marginal_draws.R
@@ -293,6 +293,7 @@ get_draws_sigma <- function(
 ) {
   draws <- tibble::as_tibble(posterior::as_draws_df(model))
   draws <- draws[, grep("^b_sigma_", colnames(draws), value = TRUE)]
+  data[[attr(data, "brm_outcome")]] <- seq_len(nrow(data))
   x <- brms::make_standata(formula = formula, data = data)$X_sigma
   colnames(x) <- paste0("b_", colnames(x))
   x <- as.data.frame(x)