FIX Explicit int cast in NN binary trees (scikit-learn#18654)

larsoner · web-flow · commit fb67c7bb00f4 · 2020-10-31T16:21:29.000+01:00
diff --git a/sklearn/neighbors/_binary_tree.pxi b/sklearn/neighbors/_binary_tree.pxi
@@ -1074,7 +1074,8 @@ cdef class BinaryTree:
         # determine number of levels in the tree, and from this
         # the number of nodes in the tree.  This results in leaf nodes
         # with numbers of points between leaf_size and 2 * leaf_size
-        self.n_levels = np.log2(fmax(1, (n_samples - 1) / self.leaf_size)) + 1
+        self.n_levels = int(
+            np.log2(fmax(1, (n_samples - 1) / self.leaf_size)) + 1)
         self.n_nodes = (2 ** self.n_levels) - 1
 
         # allocate arrays for storage
