Change cnp.float64 to float64_t

Signed-off-by: Adam Li <adam2392@gmail.com>
neurodata · Oct 17, 2023 · 09f7785 · 09f7785
1 parent 60287ae
commit 09f7785
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sklearn/tree/_tree.pxd b/sklearn/tree/_tree.pxd
@@ -107,7 +107,7 @@ cdef class BaseTree:
     ) noexcept nogil
     cdef void _compute_feature_importances(
         self,
-        cnp.float64_t[:] importances,
+        float64_t[:] importances,
         Node* node,
     ) noexcept nogil
 

diff --git a/sklearn/tree/_tree.pyx b/sklearn/tree/_tree.pyx
@@ -1506,7 +1506,7 @@ cdef class BaseTree:
         cdef float64_t normalizer = 0.
         cdef intp_t i = 0
 
-        cdef cnp.float64_t[:] importances = np.zeros(self.n_features)
+        cdef float64_t[:] importances = np.zeros(self.n_features)
 
         with nogil:
             while node != end_node:
@@ -1532,7 +1532,7 @@ cdef class BaseTree:
 
     cdef void _compute_feature_importances(
         self,
-        cnp.float64_t[:] importances,
+        float64_t[:] importances,
         Node* node
     ) noexcept nogil:
         """Compute feature importances from a Node in the Tree.