precommit fix

Author: namsaraeva

pertpy/tools/_coda/_base_coda.py

@@ -1821,7 +1821,7 @@ def plot_draw_tree(  # pragma: no cover
         h: float | None = None,
         w: float | None = None,
         dpi: int | None = 90,
-    ) -> ete3.TreeNode | ete3.TreeStyle | None:
+    ):
         """Plot a tree using input ete3 tree object.
 
         Args:
@@ -1911,7 +1911,7 @@ def plot_draw_effects(  # pragma: no cover
         h: float | None = None,
         w: float | None = None,
         dpi: int | None = 90,
-    ) -> ete3.TreeNode | ete3.TreeStyle | None:
+    ):
         """Plot a tree with colored circles on the nodes indicating significant effects with bar plots which indicate leave-level significant effects.
 
         Args:

pertpy/tools/_mixscape.py

@@ -522,7 +522,7 @@ def plot_barplot(  # pragma: no cover
         legend_text_size: int = 8,
         show: bool | None = None,
         save: bool | str | None = None,
-    ) -> Axes | None:
+    ):
         """Barplot to visualize perturbation scores calculated by the `mixscape` function.
 
         Args:
@@ -844,7 +844,7 @@ def plot_violin(  # pragma: no cover
         save: bool | str | None = None,
         ax: Axes | None = None,
         **kwargs,
-    ) -> Axes | None:
+    ):
         """Violin plot using mixscape results. Requires `pt.tl.mixscape` to be run first.
 
         Args: