Update test_features.R

michbur · Jun 19, 2024 · cb562f5 · cb562f5
1 parent 63f717d
commit cb562f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/test_features.R b/R/test_features.R
@@ -105,10 +105,10 @@ test_features <- function(target, features, criterion = "ig", adjust = "BH",
   }
 
 
-  feature_size <- if ("simple_triplet_matrix" %in% class(features)) {
+  feature_size <- if (inherits(features, "simple_triplet_matrix")) {
     slam::col_sums(features)
   } else {
-    if ("dgCMatrix" %in% class(features)) {
+    if (inherits(features, "dgCMatrix")) {
     Matrix::colSums(features)
   } else {
     colSums(features)