FIX Fixes histgradientboosting for cython 3.0 [scipy-dev] (scikit-learn#18883)

thomasjpfan · web-flow · commit 9d394c2daab6 · 2020-11-25T21:38:25.000+01:00
diff --git a/sklearn/ensemble/_hist_gradient_boosting/_bitset.pyx b/sklearn/ensemble/_hist_gradient_boosting/_bitset.pyx
@@ -57,12 +57,13 @@ def set_raw_bitset_from_binned_bitset(BITSET_INNER_DTYPE_C[:] raw_bitset,  # OUT
                                       BITSET_INNER_DTYPE_C[:] binned_bitset,
                                       X_DTYPE_C[:] categories):
     """Set the raw_bitset from the values of the binned bitset
-    
+
     categories is a mapping from binned category value to raw category value.
     """
     cdef:
-        int binned_cat_value, raw_cat_value
-    
+        int binned_cat_value
+        X_DTYPE_C raw_cat_value
+
     for binned_cat_value, raw_cat_value in enumerate(categories):
         if in_bitset_memoryview(binned_bitset, binned_cat_value):
-            set_bitset_memoryview(raw_bitset, raw_cat_value)
+            set_bitset_memoryview(raw_bitset, <X_BINNED_DTYPE_C>raw_cat_value)
