fix newdata error when family = mgcv::betar

fixes #406
fixes #407

Author: jgabry

R/stanreg-methods.R

@@ -267,14 +267,15 @@ ranef_template <- function(object) {
     new_formula <- as.formula(paste(new_formula_lhs, "~", new_formula_rhs))
   }
 
-  if (stan_fun != "stan_nlmer" && is.gaussian(object$family$family)) {
+  if (stan_fun != "stan_nlmer" && 
+      (is.gaussian(object$family$family) || is.beta(object$family$family))) {
     stan_fun <- "stan_lmer"
   }
   lme4_fun <- switch(
     stan_fun,
     "stan_lmer" = "lmer",
     "stan_nlmer" = "nlmer",
-    "glmer" # for stan_glmer, stan_glmer.nb, stan_gamm4 (unless gaussian)
+    "glmer" # for stan_glmer, stan_glmer.nb, stan_gamm4 (unless gaussian or beta)
   )
   cntrl_args <- list(optimizer = "Nelder_Mead", optCtrl = list(maxfun = 1))
   if (lme4_fun != "nlmer") { # nlmerControl doesn't allow these

tests/testthat/test_stan_glmer.R

@@ -243,6 +243,14 @@ test_that("compatible with stan_lmer with offset", {
   check_for_error(fit, offset = offs)
 })
 
+test_that("predition with family mgcv::betar doesn't error", {
+  test_data <- data.frame(y = c(0.1, 0.3), x = c(TRUE, FALSE))
+  fit <- SW(stan_glmer(y ~ (1|x), family=mgcv::betar(link="logit"), 
+                       data=test_data,  seed = 101, iter = 200, chains = 1, refresh = 0))
+  expect_silent(posterior_linpred(res, newdata=test_data))
+  expect_silent(posterior_predict(res, newdata=test_data))
+})
+
 # compare to lme4 ---------------------------------------------------------
 context("posterior_predict (compare to lme4)")
 test_that("posterior_predict close to predict.merMod for gaussian", {