From df3b668313121ec7add6068e558863cbc15c264f Mon Sep 17 00:00:00 2001
From: tonywong94 <wongt@illinois.edu>
Date: Sun, 17 May 2020 14:34:24 -0500
Subject: [PATCH] Update plot.py to pass subtree to get_lines

Change needed to allow custom coloring of the dendrogram subtrees.
---
 astrodendro/plot.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/astrodendro/plot.py b/astrodendro/plot.py
index b00124c..322432a 100644
--- a/astrodendro/plot.py
+++ b/astrodendro/plot.py
@@ -100,7 +100,7 @@ def plot_tree(self, ax, structure=None, subtree=True, autoscale=True, **kwargs):
         """
 
         # Get the lines for the dendrogram
-        lines = self.get_lines(structures=structure, **kwargs)
+        lines = self.get_lines(structures=structure, subtree=subtree, **kwargs)
 
         # Add the lines to the axes
         ax.add_collection(lines)