add sff data during treatment_effect computation

generable · Nov 18, 2024 · a533a93 · a533a93
1 parent 4b60209
commit a533a93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/R/main-treatment_effect.R b/R/main-treatment_effect.R
@@ -127,7 +127,7 @@ predict_new_subjects <- function(fit_post, fit_prior,
 
   # Pick one subject from each group
   newdat <- group_pred_input(fit_post, group_var, t_pred, time_var)
-
+  newdat <- add_sff_input(newdat, fit_post$get_model())
   # Take draws
   if (is.null(prior_param_names)) {
     prior_param_names <- id_specific_param_names(fit_post)