Update wrt main

Signed-off-by: Adam Li <[email protected]>
neurodata · Aug 8, 2024 · b201fcb · b201fcb
1 parent 971c41d
commit b201fcb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/sklearn/tree/_tree.pxd b/sklearn/tree/_tree.pxd
@@ -58,8 +58,8 @@ cdef class BaseTree:
         float64_t weighted_n_node_samples,
         uint8_t missing_go_to_left
     ) except -1 nogil
-    cdef intp_t _resize(self, intp_t capacity) except -1 nogil
-    cdef intp_t _resize_c(self, intp_t capacity=*) except -1 nogil
+    cdef int _resize(self, intp_t capacity) except -1 nogil
+    cdef int _resize_c(self, intp_t capacity=*) except -1 nogil
 
     cdef intp_t _update_node(
         self,

diff --git a/sklearn/tree/_tree.pyx b/sklearn/tree/_tree.pyx
@@ -984,7 +984,7 @@ cdef class BaseTree:
 
     Downstream classes must implement methods to actually traverse the tree.
     """
-    cdef intp_t _resize(
+    cdef int _resize(
         self,
         intp_t capacity
     ) except -1 nogil:
@@ -999,7 +999,7 @@ cdef class BaseTree:
             with gil:
                 raise MemoryError()
 
-    cdef intp_t _resize_c(self, intp_t capacity=INTPTR_MAX) except -1 nogil:
+    cdef int _resize_c(self, intp_t capacity=INTPTR_MAX) except -1 nogil:
         """Guts of _resize
 
         Returns -1 in case of failure to allocate memory (and raise MemoryError)