Merge pull request #5204 from rapidsai/branch-23.02

Forward-merge branch-23.02 to branch-23.04
rapidsai · Feb 5, 2023 · 0ae378d · 0ae378d
2 parents 9f5348e + 8f15cc1
commit 0ae378d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/cuml/dask/ensemble/randomforestclassifier.py b/python/cuml/dask/ensemble/randomforestclassifier.py
@@ -188,6 +188,10 @@ def _construct_rf(
             **kwargs
         )
 
+    @staticmethod
+    def _predict_model_on_cpu(model, X, convert_dtype):
+        return model._predict_model_on_cpu(X, convert_dtype=convert_dtype)
+
     def get_summary_text(self):
         """
         Obtain the text summary of the random forest model