Merge pull request #42 from emdann/fix-annotate-nhoods

emdann · Sep 4, 2023 · 49db7b8 · 49db7b8
2 parents af88cc5 + fa86b02
commit 49db7b8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/milopy/utils.py b/milopy/utils.py
@@ -147,7 +147,10 @@ def annotate_nhoods(adata: AnnData,
     anno_dummies = pd.get_dummies(adata.obs[anno_col])
     anno_count = adata.obsm["nhoods"].T.dot(
         scipy.sparse.csr_matrix(anno_dummies.values))
-    anno_frac = np.array(anno_count/anno_count.sum(1))
+    try:
+        anno_frac = (anno_count/anno_count.sum(1)).toarray()
+    except AttributeError: # for old version of python
+        anno_frac = np.array(anno_count/anno_count.sum(1))
 
     anno_frac = pd.DataFrame(anno_frac,
                              columns=anno_dummies.columns,