Apply suggestions from code review

Co-authored-by: Lukas Heumos <lukas.heumos@posteo.net>
scverse · Sep 6, 2024 · fc61f88 · fc61f88
1 parent 7753a9f
commit fc61f88
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/pertpy/_utils.py b/pertpy/_utils.py
@@ -14,13 +14,11 @@ def savefig_or_show(
     ext: str = "png",
 ) -> Figure | None:
     if isinstance(save, str):
-        # check whether `save` contains a figure extension
         for try_ext in [".svg", ".pdf", ".png"]:
             if save.endswith(try_ext):
                 ext = try_ext[1:]
                 save = save.replace(try_ext, "")
                 break
-        # append extension
         writekey += f"_{save}"
         save = True
 

diff --git a/pertpy/tools/_perturbation_space/_simple.py b/pertpy/tools/_perturbation_space/_simple.py
@@ -184,13 +184,13 @@ def plot_psbulk_samples(
         return_fig: bool = False,
         **kwargs,
     ) -> Figure | None:
-        """Plot the pseudobulk samples of an AnnData object. It uses Decoupler implementation.
+        """Plot the pseudobulk samples of an AnnData object.
 
         Plot the count number vs. the number of cells per pseudobulk sample.
 
         Args:
             adata: Anndata containing pseudobulk samples.
-            groupby: .obs column to color the samples by.
+            groupby: `.obs` column to color the samples by.
             {common_plot_args}
             **kwargs: Are passed to decoupler's plot_psbulk_samples.