fix predictions of spline models when using addition argument 'subset'

paul-buerkner · Sep 18, 2019 · 97c1c17 · 97c1c17
1 parent eae2fa0
commit 97c1c17
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/NEWS.md b/NEWS.md
@@ -6,6 +6,8 @@
 censored models thanks to Andrew Milne. (#744)
 * Fix error in the generated Stan code of multivariate
 `zero_inflated_binomial` models thanks to Raoul Wolf. (#756)
+* Fix predictions of spline models when using addition
+argument `subset` thanks to Ruben Arslan.
 
 # brms 2.10.0
 

diff --git a/R/data-helpers.R b/R/data-helpers.R
@@ -653,6 +653,7 @@ extract_old_standata.mvbrmsterms <- function(x, data, ...) {
 #' @export
 extract_old_standata.brmsterms <- function(x, data, ...) {
   out <- named_list(c(names(x$dpars), names(x$nlpars)))
+  data <- subset_data(data, x)
   for (dp in names(x$dpars)) {
     out[[dp]] <- extract_old_standata(x$dpars[[dp]], data, ...)
   }