remove genes w/ nan or inf values

recursionpharma · Nov 13, 2023 · 9279e34 · 9279e34
1 parent cf870cd
commit 9279e34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/efaar_benchmarking/data_loading.py b/efaar_benchmarking/data_loading.py
@@ -122,5 +122,5 @@ def load_replogle(gene_type: str, data_type: str, data_path: str = "data/") -> s
         wget.download(src, data_path + filename)
 
     adata = sc.read_h5ad(data_path + filename)
-    adata.X = adata.X[:, np.all(~adata.X.isna() & ~np.isinf(adata.X), axis=0)]
+    adata.X = adata.X[:, np.all(~np.isnan(adata.X) & ~np.isinf(adata.X), axis=0)]
     return adata