From fc6bb8480a41e3468fe4bd985e8ed674b9ebb280 Mon Sep 17 00:00:00 2001 From: Gabriel Daiha Date: Fri, 10 May 2024 13:59:05 -0300 Subject: [PATCH] fixed criterion.pxd with new signature Signed-off-by: Gabriel Daiha --- econml/tree/_criterion.pxd | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/econml/tree/_criterion.pxd b/econml/tree/_criterion.pxd index 2eb6b8ba3..c0125d788 100644 --- a/econml/tree/_criterion.pxd +++ b/econml/tree/_criterion.pxd @@ -78,7 +78,9 @@ cdef class Criterion: cdef void node_value(self, double* dest) nogil cdef void node_jacobian(self, double* dest) nogil cdef void node_precond(self, double* dest) nogil - cdef double impurity_improvement(self, double impurity) nogil + cdef double impurity_improvement(self, double impurity_parent, + double impurity_left, + double impurity_right) nogil cdef double proxy_impurity_improvement(self) nogil cdef double min_eig_left(self) nogil cdef double min_eig_right(self) nogil