From 2eebb764bcfc1f6623b904be1bf91ef8ae3e7d79 Mon Sep 17 00:00:00 2001
From: WANG Zhiwei <48282751+statwangz@users.noreply.github.com>
Date: Sat, 16 Sep 2023 01:02:57 +0800
Subject: [PATCH] Update mfairInitialization.R

---
 R/mfairInitialization.R | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/R/mfairInitialization.R b/R/mfairInitialization.R
index c732bd9..c58a0e8 100644
--- a/R/mfairInitialization.R
+++ b/R/mfairInitialization.R
@@ -2,7 +2,7 @@
 #'
 #' @importFrom methods new
 #'
-#' @param Y A matrix. The main data matrix of N samples and M features.
+#' @param Y A matrix or Matrix::dgCMatrix. The main data matrix of N samples and M features.
 #' @param X A data.frame. The auxiliary information data frame of N samples and C covariates.
 #' @param Y_center Logical. Determines whether centering is performed.
 #' @param K_max An integer. The maximum rank allowed in the model.
@@ -11,6 +11,10 @@
 #' @return Returns MFAIR object, with main data matrix and auxiliary information.
 #' @export
 createMFAIR <- function(Y, X, Y_center = TRUE, K_max = 1L, project = "MFAIR") {
+  if(is.data.frame(X)){
+    stop("X should be a data.frame!")
+  } # End
+
   # Data dimension
   N <- nrow(Y)
   M <- ncol(Y)