added plots dir

YosefLab · May 14, 2024 · bead9cd · bead9cd
1 parent b9eb11c
commit bead9cd
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -28,4 +28,7 @@ __pycache__/
 /.vscode/
 
 # Plots
-/tests/plots/*.png
+/tests/plots/*.png
+
+# Node modules
+node_modules/
diff --git a/src/pycea/pl/tree.py b/src/pycea/pl/tree.py
@@ -1,5 +1,6 @@
+from __future__ import annotations
+
 from collections.abc import Mapping, Sequence
-from typing import Union
 
 import cycler
 import matplotlib.colors as mcolors
@@ -25,8 +26,8 @@ def branches(
     angled_branches: bool = False,
     color: str = "black",
     color_na: str = "lightgrey",
-    linewidth: Union[int, float, str] = 1,
-    linewidth_na: Union[int, float] = 1,
+    linewidth: int | float | str = 1,
+    linewidth_na: int | float = 1,
     cmap: str | mcolors.Colormap = "viridis",
     palette: cycler.Cycler | mcolors.ListedColormap | Sequence[str] | Mapping[str] | None = None,
     ax: Axes | None = None,

diff --git a/tests/plots/readme.md b/tests/plots/readme.md
@@ -0,0 +1 @@
+Folder for storing plots and figures.