fix dtypes for clustering

nih-fmrif · Jan 8, 2024 · 7c5df8d · 7c5df8d
1 parent 9764d02
commit 7c5df8d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/contarg/clustering.py b/contarg/clustering.py
@@ -146,7 +146,7 @@ def repdist_tree(X, X_for_connected_components, connectivity,
     if return_distance:
         distances = np.empty(n_nodes - n_samples)
 
-    not_visited = np.empty(n_nodes, dtype=np.int8, order="C")
+    not_visited = np.empty(n_nodes, dtype=bool, order="C")
 
     # recursive merge loop
     for k in range(n_samples, n_nodes):