mids fix

istallworthy · Oct 9, 2023 · 8bf693c · 8bf693c
1 parent fc8d3fa
commit 8bf693c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/R/createWeights.R b/R/createWeights.R
@@ -93,10 +93,10 @@ createWeights <- function(home_dir, data, exposure, outcome, formulas, method =
     stop ("Please provide either a 'mids' object, a data frame, or a list of imputed data frames in the 'data' field.",
          call. = FALSE)
   }
-  else if (is.list(data) && !is.data.frame(data)) {
+  else if (is.list(data) && !is.data.frame(data)  && !mice::is.mids(data)) {
     if (sum(sapply(data, is.data.frame)) != length(data)) {
       stop ("Please supply a list of data frames that have been imputed.",
-           call. = FALSE)
+            call. = FALSE)
     }
   }
 

diff --git a/R/fitModel.R b/R/fitModel.R
@@ -129,7 +129,7 @@ fitModel <- function(home_dir, data, weights, exposure, exposure_time_pts, outco
     stop ("Please provide either a 'mids' object, a data frame, or a list of imputed data frames in the 'data' field.",
           call. = FALSE)
   }
-  else if (is.list(data) && !is.data.frame(data)) {
+  else if (is.list(data) && !is.data.frame(data)  && !mice::is.mids(data)) {
     if (sum(sapply(data, is.data.frame)) != length(data)) {
       stop ("Please supply a list of data frames that have been imputed.",
             call. = FALSE)