Merge pull request #200 from raphael-group/pandasdepr

pandas groupby depreciation fix
raphael-group · Apr 3, 2024 · 369979c · 369979c
2 parents cce2869 + c2e6eac
commit 369979c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/hatchet/utils/plot_cn_1d2d.py b/src/hatchet/utils/plot_cn_1d2d.py
@@ -390,7 +390,7 @@ def plot_genome(
                     )
 
             if n_clones == 1:
-                my_colors = [cmap(mapping[r]) for r in bbc.cn_clone1]
+                my_colors = [cmap(mapping[tuple([r])]) for r in bbc.cn_clone1]
             else:
                 my_colors = [
                     cmap(mapping[tuple(r)]) for _, r in bbc[[f'cn_clone{i + 1}' for i in range(n_clones)]].iterrows()
@@ -523,7 +523,7 @@ def plot_clusters(
         assert n_clones2 == n_clones, (n_clones2, n_clones)
 
         if n_clones == 1:
-            my_colors = [cmap(mapping[r]) for r in bbc_.cn_clone1]
+            my_colors = [cmap(mapping[tuple([r])]) for r in bbc_.cn_clone1]
         else:
             my_colors = [
                 cmap(mapping[tuple(r)]) for _, r in bbc_[[f'cn_clone{i + 1}' for i in range(n_clones)]].iterrows()