Add check for NA, NaN, Inf values in cell attributes (see issue #85)

satijalab · Dec 15, 2020 · ad50748 · ad50748
1 parent 07f41f3
commit ad50748
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/R/utils.R b/R/utils.R
@@ -39,6 +39,16 @@ make_cell_attr <- function(umi, cell_attr, latent_var, batch_var, latent_var_non
     new_attr <- do.call(cbind, new_attr)
     cell_attr <- cbind(cell_attr, new_attr[, setdiff(colnames(new_attr), colnames(cell_attr)), drop = TRUE])
   }
+
+  # make sure no NA, NaN, Inf values are in cell attributes - they would cause
+  # problems later on
+  rel_attr <- cell_attr[, c(latent_var, batch_var, latent_var_nonreg)]
+  if (any(is.na(rel_attr)) || 
+      any(is.nan(rel_attr)) || 
+      any(is.infinite(rel_attr))) {
+    stop('cell attributes cannot contain any NA, NaN, or infinite values')
+  }
+
   return(cell_attr)
 }