Merge branch 'main' of github.com:cwindolf/dartsort

cwindolf · Dec 6, 2024 · 535a2db · 535a2db
2 parents fb2e652 + 6e1da32
commit 535a2db
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/dartsort/vis/analysis_plots.py b/src/dartsort/vis/analysis_plots.py
@@ -53,6 +53,7 @@ def annotated_dendro(
     above_threshold_color=0.0,
     leaf_labels=None,
     annotations_offset_by_n=True,
+    rotation=45,
 ):
     n = len(Z) + 1
     res = dendrogram(
@@ -69,7 +70,7 @@ def annotated_dendro(
     for j, (ic, dc) in enumerate(zip(res["icoord"], res["dcoord"])):
         lines[j, :, 0] = ic
         lines[j, :, 1] = dc
-        if dc[1] < 1.0:
+        if dc[1] < threshold:
             colors[j] = glasbey1024[n + j]
         else:
             colors[j] = above_threshold_color
@@ -99,7 +100,9 @@ def annotated_dendro(
                 color=lc,
                 va="center",
                 ha="center",
-                bbox=dict(fc=fc, ec="none", boxstyle="square,pad=0.1"),
+                rotation=rotation,
+                rotation_mode="anchor",
+                bbox=dict(fc=fc, ec="none", boxstyle="square,pad=0."),
             )