Fix BaseRandomForestModel.

rapidsai · Jan 31, 2023 · c171cae · c171cae
1 parent 9023aa1
commit c171cae
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/cuml/ensemble/randomforest_common.pyx b/python/cuml/ensemble/randomforest_common.pyx
@@ -290,6 +290,7 @@ class BaseRandomForestModel(Base):
                 y_m, cp.arange(self.num_classes, dtype=np.int32))
             if self.use_monotonic:
                 y_m, _ = make_monotonic(y_m)
+                y_m = input_to_cuml_array(y_m).array
 
         else:
             y_m, _, _, y_dtype = \