fix

strengejacke · strengejacke · commit 850ed8c83bef · 2025-11-14T14:00:34.000+01:00
diff --git a/R/check_predictions.R b/R/check_predictions.R
@@ -226,6 +226,11 @@ check_predictions.stanreg <- function(
     )
   }
 
+  # make x cateogorical for bernoulli/categorical/multinomial models
+  if (minfo$is_bernoulli || minfo$is_categorical || minfo$is_multinomial) {
+    out$x <- as.factor(out$x)
+  }
+
   attr(out, "is_stan") <- TRUE
   attr(out, "check_range") <- check_range
   attr(out, "response_name") <- resp_string

Original file line number	Diff line number	Diff line change
`@@ -226,6 +226,11 @@ check_predictions.stanreg <- function(`
`226`	`226`	`)`
`227`	`227`	`}`
`228`	`228`
	`229`	`+ # make x cateogorical for bernoulli/categorical/multinomial models`
	`230`	`+ if (minfo$is_bernoulli \|\| minfo$is_categorical \|\| minfo$is_multinomial) {`
	`231`	`+ out$x <- as.factor(out$x)`
	`232`	`+ }`
	`233`	`+`
`229`	`234`	`attr(out, "is_stan") <- TRUE`
`230`	`235`	`attr(out, "check_range") <- check_range`
`231`	`236`	`attr(out, "response_name") <- resp_string`