Revert mixscape plots

Signed-off-by: zethson <lukas.heumos@posteo.net>
scverse · Apr 6, 2024 · b4863f9 · b4863f9
1 parent 14d1e27
commit b4863f9
Showing 1 changed file with 0 additions and 26 deletions.
diff --git a/pertpy/tools/_mixscape.py b/pertpy/tools/_mixscape.py
@@ -613,8 +613,6 @@ def plot_heatmap(  # pragma: no cover
         subsample_number: int | None = 900,
         vmin: float | None = -2,
         vmax: float | None = 2,
-        fontsize: int | None = 8,
-        ax: Axes | None = None,
         show: bool | None = None,
         save: bool | str | None = None,
         **kwds,
@@ -667,8 +665,6 @@ def plot_heatmap(  # pragma: no cover
             vmax=vmax,
             n_genes=20,
             groups=["NT"],
-            fontsize=fontsize,
-            ax=ax,
             show=show,
             save=save,
             **kwds,
@@ -775,17 +771,6 @@ def plot_perturbscore(  # pragma: no cover
                 plt.legend(title="gene_target", title_fontsize=14, fontsize=12)
                 sns.despine()
 
-        ax = plt.gca()
-
-        if save:
-            plt.savefig(save, bbox_inches="tight")
-            return None
-        if show:
-            plt.show()
-            return None
-        elif not show or show is None:
-            return ax
-
         # If before_mixscape is False, split densities based on mixscape classifications
         else:
             if palette is None:
@@ -842,17 +827,6 @@ def plot_perturbscore(  # pragma: no cover
                 plt.legend(title="mixscape class", title_fontsize=14, fontsize=12)
                 sns.despine()
 
-        ax = plt.gca()
-
-        if save:
-            plt.savefig(save, bbox_inches="tight")
-            return None
-        if show:
-            plt.show()
-            return None
-        elif not show or show is None:
-            return ax
-
     def plot_violin(  # pragma: no cover
         self,
         adata: AnnData,