fix matrix check issue

uclahs-cds · Aug 23, 2024 · 4d797db · 4d797db
1 parent 8423163
commit 4d797db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/create.autoencoder.R b/R/create.autoencoder.R
@@ -8,8 +8,8 @@ create.autoencoder <- function(
 	) {
 
 	# input checks
-	if ('matrix' != class(data.matrix) && 'data.frame' != class(data.matrix)) {
-		stop('data.matrix needs to be a matrix or a data frame');
+	if (!("matrix" %in% class(data.matrix) || "data.frame" %in% class(data.matrix))) {
+		stop('data.matrix needs to be a matrix or a data frame')
 		}
 	if (any(is.na(data.matrix))) {
 		stop('data matrix contains NA(s)');