detect panelr as mixed model

easystats · Jul 24, 2024 · 86d71c8 · 86d71c8
1 parent 70f7e04
commit 86d71c8
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/NAMESPACE b/NAMESPACE
@@ -1048,6 +1048,8 @@ S3method(is_converged,merMod)
 S3method(is_mixed_model,afex_aov)
 S3method(is_mixed_model,default)
 S3method(is_mixed_model,marginaleffects)
+S3method(is_mixed_model,wbgee)
+S3method(is_mixed_model,wbm)
 S3method(is_nullmodel,afex_aov)
 S3method(is_nullmodel,default)
 S3method(link_function,BBmm)

diff --git a/R/is_mixed_model.R b/R/is_mixed_model.R
@@ -35,3 +35,11 @@ is_mixed_model.afex_aov <- function(x) {
 is_mixed_model.marginaleffects <- function(x) {
   is_mixed_model(attributes(x)$model)
 }
+
+#' @export
+is_mixed_model.wbm <- function(x) {
+  TRUE
+}
+
+#' @export
+is_mixed_model.wbgee <- is_mixed_model.wbm
diff --git a/tests/testthat/test-get_variance.R b/tests/testthat/test-get_variance.R
@@ -38,7 +38,7 @@ v6 <- suppressWarnings(get_variance(fm6))
 test_that("error for non-mixed", {
   data(mtcars)
   expect_error(
-    get_vaiance(lm(mpg ~ gear, data = mtcars)),
+    get_variance(lm(mpg ~ gear, data = mtcars)),
     regex = "This function only works for mixed models"
   )
 })