Ensure int casts in condensed tree scikit-learn-contrib#600

healthtechconnex1 · Aug 31, 2023 · f1833db · f1833db
1 parent 70bc4d5
commit f1833db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hdbscan/_hdbscan_tree.pyx b/hdbscan/_hdbscan_tree.pyx
@@ -106,12 +106,12 @@ cpdef np.ndarray condense_tree(np.ndarray[np.double_t, ndim=2] hierarchy,
             lambda_value = INFTY
 
         if left >= num_points:
-            left_count = <np.intp_t> hierarchy[left - num_points][3]
+            left_count = <np.intp_t> int(hierarchy[left - num_points][3])
         else:
             left_count = 1
 
         if right >= num_points:
-            right_count = <np.intp_t> hierarchy[right - num_points][3]
+            right_count = <np.intp_t> int(hierarchy[right - num_points][3])
         else:
             right_count = 1