Fix for #11

TutteInstitute · Feb 25, 2024 · cd45c0d · cd45c0d
1 parent 458d2ad
commit cd45c0d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/datamapplot/create_plots.py b/datamapplot/create_plots.py
@@ -529,6 +529,8 @@ def create_interactive_plot(
         )
         for labels in reversed(label_layers):
             label_map = {n: i for i, n in enumerate(np.unique(labels))}
+            if noise_label not in label_map:
+                label_map[noise_label] = -1
             label_unmap = {i: n for n, i in label_map.items()}
             cluster_label_vector = np.asarray(pd.Series(labels).map(label_map))
             unique_non_noise_labels = [