fixed criterion.pxd with new signature

Signed-off-by: Gabriel Daiha <[email protected]>
py-why · May 10, 2024 · fc6bb84 · fc6bb84
1 parent c0c5e6d
commit fc6bb84
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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