From b61ae3d546ba4199dc3badf4bd89971d2d75e9df Mon Sep 17 00:00:00 2001 From: Adam Li Date: Sat, 9 Mar 2024 22:47:53 -0500 Subject: [PATCH] Export shift_missing_values_to_left_if_required Signed-off-by: Adam Li --- sklearn/tree/_splitter.pxd | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/sklearn/tree/_splitter.pxd b/sklearn/tree/_splitter.pxd index 601e6ac8f3202..041e9965a904b 100644 --- a/sklearn/tree/_splitter.pxd +++ b/sklearn/tree/_splitter.pxd @@ -140,3 +140,9 @@ cdef class Splitter(BaseSplitter): float64_t lower_bound, float64_t upper_bound ) noexcept nogil + +cdef void shift_missing_values_to_left_if_required( + SplitRecord* best, + intp_t[::1] samples, + intp_t end, +) noexcept nogil