fix issue #1540

paul-buerkner · Sep 14, 2023 · 3e47553 · 3e47553
1 parent 2155a6f
commit 3e47553
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/DESCRIPTION b/DESCRIPTION
@@ -2,8 +2,8 @@ Package: brms
 Encoding: UTF-8
 Type: Package
 Title: Bayesian Regression Models using 'Stan'
-Version: 2.20.2
-Date: 2023-09-09
+Version: 2.20.3
+Date: 2023-09-14
 Authors@R:
     c(person("Paul-Christian", "Bürkner", email = "paul.buerkner@gmail.com",
              role = c("aut", "cre")),

diff --git a/R/predictor.R b/R/predictor.R
@@ -59,7 +59,12 @@ predictor.bprepnl <- function(prep, i = NULL, fprep = NULL, ...) {
     # when 'nlform' must be evaluated jointly across observations
     # and hence 'loop' had been set to FALSE
     for (i in seq_along(args)) {
-      args[[i]] <- split(args[[i]], row(args[[i]]))
+      old_dim <- dim(args[[i]])
+      args[[i]] <- split(args[[i]], slice.index(args[[i]], 1))
+      if (length(old_dim) > 2L) {
+        # split drops array dimensions which need to be restored
+        args[[i]] <- lapply(args[[i]], "dim<-", old_dim[-1])
+      }
     }
     .fun <- function(...) eval(prep$nlform, list(...))
     eta <- try(

diff --git a/man/save_pars.Rd b/man/save_pars.Rd