fix size error in plot_da_beeswarm (#551)

Co-authored-by: EC2 Default User <ec2-user@ip-172-31-19-17.eu-west-2.compute.internal>
scverse · Mar 6, 2024 · b51a2c9 · b51a2c9
1 parent 44af066
commit b51a2c9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/pertpy/plot/_milopy.py b/pertpy/plot/_milopy.py
@@ -139,7 +139,7 @@ def da_beeswarm(
 
         Args:
             mdata: MuData object
-            anno_col: Column in adata.uns['nhood_adata'].obs to use as annotation. (default: 'nhood_annotation'.)
+            anno_col: Column in mdata['milo'].var to use as annotation. (default: 'nhood_annotation'.)
             alpha: Significance threshold. (default: 0.1)
             subset_nhoods: List of nhoods to plot. If None, plot all nhoods. (default: None)
             palette: Name of Seaborn color palette for violinplots.

diff --git a/pertpy/tools/_milo.py b/pertpy/tools/_milo.py
@@ -861,7 +861,7 @@ def plot_da_beeswarm(
             nhood_adata.obs[anno_col]
         except KeyError:
             raise RuntimeError(
-                f"Unable to find {anno_col} in mdata.uns['nhood_adata']. Run 'milopy.utils.annotate_nhoods(adata, anno_col)' first"
+                f"Unable to find {anno_col} in mdata['milo'].var. Run 'milopy.utils.annotate_nhoods(adata, anno_col)' first"
             ) from None
 
         if subset_nhoods is not None:
@@ -897,7 +897,6 @@ def plot_da_beeswarm(
                 y=anno_col,
                 x="logFC",
                 order=sorted_annos,
-                size=190,
                 inner=None,
                 orient="h",
                 palette=palette,
@@ -910,7 +909,6 @@ def plot_da_beeswarm(
                 y=anno_col,
                 x="logFC",
                 order=sorted_annos,
-                size=190,
                 inner=None,
                 orient="h",
                 linewidth=0,