Fix issue #243 (#262)

stan-dev · Jan 8, 2022 · b894b36 · b894b36
1 parent 3392c86
commit b894b36
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/R/cv_varsel.R b/R/cv_varsel.R
@@ -394,7 +394,7 @@ loo_varsel <- function(refmodel, method, nterms_max, ndraws,
     for (k in seq_along(submodels)) {
       mu_k <- refmodel$family$mu_fun(submodels[[k]]$sub_fit,
                                      obs = inds,
-                                     offset = refmodel$offset)
+                                     offset = refmodel$offset[inds])
       log_lik_sub <- t(refmodel$family$ll_fun(
         mu_k, submodels[[k]]$dis, refmodel$y[inds], refmodel$wobs[inds]
       ))

diff --git a/R/refmodel.R b/R/refmodel.R
@@ -567,6 +567,8 @@ init_refmodel <- function(object, data, formula, family, ref_predfun = NULL,
     newdata <- fetch_data(data, obs = obs, newdata = newdata)
     if (is.null(offset)) {
       offset <- rep(0, nrow(newdata))
+    } else {
+      stopifnot(length(offset) %in% c(1L, nrow(newdata)))
     }
     family$linkinv(proj_predfun(fit, newdata = newdata) + offset)
   }