Update train.py

szawan · Apr 23, 2023 · 3044baa · 3044baa
1 parent 975132e
commit 3044baa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TOSICA/train.py b/TOSICA/train.py
@@ -30,7 +30,7 @@ def set_seed(seed):
 
 def todense(adata):
     import scipy
-    if isinstance(adata.X, scipy.sparse.csr_matrix):
+    if isinstance(adata.X, scipy.sparse.csr_matrix) or isinstance(adata.X, scipy.sparse.csc_matrix):
         return adata.X.todense()
     else:
         return adata.X