8000 MAINT Fix C warning in Cython module splitting.pyx · scikit-learn/scikit-learn@6d173c0 · GitHub
[go: up one dir, main page]

Skip to content

Commit 6d173c0

Browse files
committed
MAINT Fix C warning in Cython module splitting.pyx
1 parent de968ed commit 6d173c0

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

sklearn/ensemble/_hist_gradient_boosting/splitting.pyx

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -569,15 +569,15 @@ cdef class Splitter:
569569
free(split_infos)
570570
return out
571571

572-
cdef unsigned int _find_best_feature_to_split_helper(
572+
cdef int _find_best_feature_to_split_helper(
573573
self,
574574
split_info_struct * split_infos, # IN
575575
int n_allowed_features,
576576
) noexcept nogil:
577577
"""Return the index of split_infos with the best feature split."""
578578
cdef:
579-
unsigned int split_info_idx
580-
unsigned int best_split_info_idx = 0
579+
int split_info_idx
580+
int best_split_info_idx = 0
581581

582582
for split_info_idx in range(1, n_allowed_features):
583583
if (split_infos[split_info_idx].gain > split_infos[best_split_info_idx].gain):

0 commit comments

Comments
 (0)
0