Skip to content

Commit b15152e

Browse files
committed
update DendrogramPlotter
1 parent 45813ca commit b15152e

2 files changed

Lines changed: 4 additions & 2 deletions

File tree

PyComplexHeatmap/annotations.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1440,7 +1440,9 @@ def _type_specific_params(self):
14401440
pass
14411441

14421442
def plot(self, ax=None, axis=1):
1443-
self.plot_kws.setdefault("tree_kws", dict(colors=self.colors))
1443+
if 'tree_kws' not in self.plot_kws:
1444+
self.plot_kws['tree_kws']={}
1445+
self.plot_kws["tree_kws"].setdefault(colors=self.colors)
14441446
# inint the DendrogramPlotter class object
14451447
ax.set_axis_off()
14461448
self.dend.plot(ax=ax,axis=axis,**self.plot_kws)

PyComplexHeatmap/clustermap.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -986,7 +986,7 @@ def plot(self, ax, gap_pixel=None, root_x=None,tree_kws=None,
986986
bx, by = evaluate_bezier(getControlPoints((x[0],y[0]),center,axis=self.axis))
987987
ax.plot(bx, by, color=color)
988988
bx, by = evaluate_bezier(getControlPoints((x[-1], y[-1]), center,axis=self.axis))
989-
ax.plot(bx, by, color=color)
989+
ax.plot(bx, by, color=color,**tree_kws)
990990
if root_dot:
991991
ax.scatter(center_x, center_y, color=color,s=dotsize)
992992
ax.scatter(x[0], y[0], color=color, s=dotsize)

0 commit comments

Comments
 (0)