[DRAFT] adding support for the {mmrm} package

DESCRIPTION

@@ -112,6 +112,7 @@ Suggests:
     miceadds,
     mlogit,
     mlr3verse,
+    mmrm (>= 0.3.12),
     modelbased,
     modelsummary,
     nlme,
@@ -246,6 +247,7 @@ Collate:
     'methods_mlm.R'
     'methods_mlogit.R'
     'methods_mlr3.R'
+    'methods_mmrm.R'
     'methods_nlme.R'
     'methods_ordinal.R'
     'methods_plm.R'

NAMESPACE

@@ -21,6 +21,7 @@ S3method(get_coef,lmerModLmerTest)
 S3method(get_coef,mblogit)
 S3method(get_coef,merMod)
 S3method(get_coef,mlm)
+S3method(get_coef,mmrm)
 S3method(get_coef,multinom)
 S3method(get_coef,nls)
 S3method(get_coef,polr)
@@ -76,6 +77,7 @@ S3method(get_predict,mblogit)
 S3method(get_predict,merMod)
 S3method(get_predict,mhurdle)
 S3method(get_predict,mlogit)
+S3method(get_predict,mmrm)
 S3method(get_predict,model_fit)
 S3method(get_predict,multinom)
 S3method(get_predict,ols)
@@ -104,6 +106,7 @@ S3method(get_vcov,glimML)
 S3method(get_vcov,glmmTMB)
 S3method(get_vcov,gsm)
 S3method(get_vcov,mhurdle)
+S3method(get_vcov,mmrm)
 S3method(get_vcov,model_fit)
 S3method(get_vcov,orm)
 S3method(get_vcov,pstpm2)
@@ -160,6 +163,7 @@ S3method(set_coef,lmerMod)
 S3method(set_coef,lmerModLmerTest)
 S3method(set_coef,merMod)
 S3method(set_coef,mlm)
+S3method(set_coef,mmrm)
 S3method(set_coef,model_fit)
 S3method(set_coef,multinom)
 S3method(set_coef,nls)

NEWS.md

@@ -9,9 +9,10 @@ Breaking changes:
 
 New models supported:
 
-* `stpm2`, `pstpm2`, `gsm`, and `aft` models from `rstpm2`. Thanks to @aghaynes and @mclements.
+* `rstpm2` package: support for `stpm2`, `pstpm2`, `gsm`, and `aft` models. Thanks to @mclements for contribution #1188 and @aghayes for useful comments.
 * `glm_weightit`, `coxph_weightit`, `multinom_weightit`, and `ordinal_weightit` models from `Weightit`. Thanks to @ngreifer.
 * `glmmgee` from the `glmtoolbox` package. Thanks to @adrianolszewski for the request and @lhvanegasp for help with implementation.
+* `mmrm` package for mixed effects modelling. Thanks to @kkmann for contribution #1000, as well as @clarkliming and @lang-benjamin for useful comments.
 
 New features:
 

R/methods_mmrm.R

@@ -0,0 +1,48 @@
+#' @include set_coef.R
+#' @rdname set_coef
+#' @keywords internal
+#' @export
+set_coef.mmrm <- function(model, coefs, ...) {
+  model$beta_est  <- coefs
+  return(model)
+}
+
+
+#' @include get_coef.R
+#' @rdname get_coef
+#' @keywords internal
+#' @export
+get_coef.mmrm <- function(model, ...) {
+  return(coef(model, ...))
+}
+
+
+#' @include get_vcov.R
+#' @rdname get_vcov
+#' @keywords internal
+#' @export
+get_vcov.mmrm <- function(model, ...) {
+  return(vcov(model, ...))
+}
+
+
+#' @rdname get_predict
+#' @keywords internal
+#' @export
+get_predict.mmrm <- function(model, newdata = model$data, type = "response", ...) {
+  type <- match.arg(type)
+  res <- data.frame(
+    rowid = seq_len(nrow(newdata)),
+    estimate = predict(model, newdata = newdata, type = type, conditional = FALSE)
+  )
+  return(res)
+}
+
+
+#' @include sanity_model.R
+#' @rdname sanitize_model_specific
+#' @keywords internal
+sanitize_model_specific.mmrm <- function(model, ...) {
+    insight::check_if_installed("mmrm", minimum_version = "0.3.11.9000")
+    return(model)
+}

R/sanity_model.R

@@ -88,6 +88,7 @@ sanity_model_supported_class <- function(model) {
         "mhurdle",
         "mira",
         "mlogit",
+        "mmrm",
         "model_fit",
         c("multinom", "nnet"),
         "multinom_weightit",

inst/tinytest/test-pkg-mmrm.R

@@ -0,0 +1,17 @@
+source("helpers.R")
+using("marginaleffects")
+
+requiet("mmrm")
+
+fit <- mmrm(
+  formula = FEV1 ~ RACE + SEX + ARMCD * AVISIT + us(AVISIT | USUBJID),
+  data = fev_data
+)
+
+pre <- avg_predictions(fit, newdata = fev_data)
+expect_inherits(pre, "predictions")
+
+cmp <- avg_comparisons(fit, 
+    variables = "SEX",
+    newdata = fev_data)
+expect_inherits(cmp, "comparisons")