diff --git a/NAMESPACE b/NAMESPACE index f0f42c97c..c328b677a 100644 --- 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 index e3d9cb3d6..c3df0bb0a 100644 --- 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 index 637f15322..261e0ab2b 100644 --- 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" ) })