Update funs.py

Polkas · Oct 4, 2023 · 6684a1e · 6684a1e
1 parent 211bfd6
commit 6684a1e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/multidim/funs.py b/src/multidim/funs.py
@@ -104,7 +104,7 @@ def plot_dendrogram(model: Any, **kwargs: Optional[Any]) -> None:
                 current_count += counts[child_idx - n_samples]
         counts[i] = current_count
 
-    linkage_matrix = np.column_stack(
+    linkage_matrix: np.ndarray = np.column_stack(
         [model.children_, model.distances_, counts]
     ).astype(float)