Fix use of subtree parameter in plot_tree and get_mask

tonywong94 · Aug 18, 2022 · b4d0261 · b4d0261
1 parent 75ec031
commit b4d0261
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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)

diff --git a/astrodendro/structure.py b/astrodendro/structure.py
@@ -477,7 +477,7 @@ def get_mask(self, shape=None, subtree=True):
         """
         if shape is None:
             shape = self._dendrogram.data.shape
-        indices = self.indices(subtree=True) if subtree else self.indices
+        indices = self.indices(subtree=True) if subtree else self.indices()
         mask = np.zeros(shape, dtype=bool)
         mask[indices] = True
         return mask