Skip to content

Commit b201fcb

Browse files
committed
Update wrt main
Signed-off-by: Adam Li <adam2392@gmail.com>
1 parent 971c41d commit b201fcb

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

sklearn/tree/_tree.pxd

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,8 +58,8 @@ cdef class BaseTree:
5858
float64_t weighted_n_node_samples,
5959
uint8_t missing_go_to_left
6060
) except -1 nogil
61-
cdef intp_t _resize(self, intp_t capacity) except -1 nogil
62-
cdef intp_t _resize_c(self, intp_t capacity=*) except -1 nogil
61+
cdef int _resize(self, intp_t capacity) except -1 nogil
62+
cdef int _resize_c(self, intp_t capacity=*) except -1 nogil
6363

6464
cdef intp_t _update_node(
6565
self,

sklearn/tree/_tree.pyx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -984,7 +984,7 @@ cdef class BaseTree:
984984
985985
Downstream classes must implement methods to actually traverse the tree.
986986
"""
987-
cdef intp_t _resize(
987+
cdef int _resize(
988988
self,
989989
intp_t capacity
990990
) except -1 nogil:
@@ -999,7 +999,7 @@ cdef class BaseTree:
999999
with gil:
10001000
raise MemoryError()
10011001

1002-
cdef intp_t _resize_c(self, intp_t capacity=INTPTR_MAX) except -1 nogil:
1002+
cdef int _resize_c(self, intp_t capacity=INTPTR_MAX) except -1 nogil:
10031003
"""Guts of _resize
10041004
10051005
Returns -1 in case of failure to allocate memory (and raise MemoryError)

0 commit comments

Comments
 (0)