eaano: matrix function can return a simple vector or data.frame

Yunuuuu · Jun 26, 2024 · 029f973 · 029f973
1 parent 0172cd4
commit 029f973
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/R/eanno.R b/R/eanno.R
@@ -310,9 +310,15 @@ methods::setMethod(
  row = heat_matrix,
  column = t(heat_matrix)
  )
- if (!is.matrix(mat <- mat(data))) {
- cli::cli_abort("{.fn @matrix} of {id} must return a matrix")
- }
+ mat <- tryCatch(
+ build_matrix(mat(data)),
+ function(cnd) {
+ cli::cli_abort(paste(
+ "{.fn @matrix} of {id} must return a {.cls matrix},",
+ "a simple vector, or a {.cls data.frame}."
+ ))
+ }
+ )
  object@n <- nrow(mat)
  }
  object@matrix <- mat