Merge pull request TutteInstitute#31 from JelmerBot/dev/condensed_plot

fix plotting fast_hdbscan condensed trees

Author: lmcinnes

doc/detecting_branches.ipynb

@@ -410,8 +410,7 @@ def condensed_tree_(self):
         if self._condensed_tree is not None:
             return CondensedTree(
                 self._condensed_tree,
-                self.cluster_selection_method,
-                self.allow_single_cluster,
+                self.labels_,
             )
         else:
             raise AttributeError(

fast_hdbscan/hdbscan.py

@@ -515,8 +515,8 @@ def propagated_labels(self):
             msg="You first need to fit the SubClusterDetector model before propagating the labels.",
         )
         return propagate_sub_cluster_labels(
-            self.labels_,
-            self.sub_cluster_labels_,
+            self.labels_.copy(),
+            self.sub_cluster_labels_.copy(),
             self._approximation_graphs,
             self.cluster_points_,
         )

fast_hdbscan/sub_clusters.py

@@ -88,8 +88,12 @@ def test_selection_method():
 def test_label_propagation():
     b = SubClusterDetector(lens_values=centrality).fit(c)
     check_detected_groups(b, n_clusters=2, n_subs=7)
-    b.labels_, b.sub_cluster_labels_ = b.propagated_labels()
-    assert np.all(b.sub_cluster_labels_ >= 0)
+    prev_sub_labels = b.sub_cluster_labels_.copy()
+    labels_, sub_cluster_labels_ = b.propagated_labels()
+    assert np.all(b.sub_cluster_labels_ == prev_sub_labels)
+    assert np.all(sub_cluster_labels_ >= 0)
+    b.labels_ = labels_
+    b.sub_cluster_labels_ = sub_cluster_labels_
     check_detected_groups(b, n_clusters=2, n_subs=5)
 
     b = SubClusterDetector(lens_values=centrality, propagate_labels=True).fit(c)