Adapt to new format of failed saem.mmkin fits

1 files changed, 9 insertions, 7 deletions

diff --git a/R/mhmkin.R b/R/mhmkin.R
index a1475ef9..17c8a1f7 100644
--- a/R/mhmkin.R
+++ b/R/mhmkin.R
@@ -140,13 +140,15 @@ print.mhmkin <- function(x, ...) {
 convergence.mhmkin <- function(object, ...) {
   all_summary_warnings <- character()
 
-  result <- lapply(object,
-    function(fit) {
-      if (inherits(fit, "try-error")) return("E")
-      else {
-        return("OK")
-      }
-  })
+  if (inherits(object[[1]], "saem.mmkin")) {
+    test_func <- function(fit) {
+      if (inherits(fit$so, "try-error")) return("E")
+      else return("OK")
+    }
+  } else {
+    stop("Only mhmkin objects containing saem.mmkin objects currently supported")
+  }
+  result <- lapply(object, test_func)
   result <- unlist(result)
   dim(result) <- dim(object)
   dimnames(result) <- dimnames(object)