diff --git a/python/cuml/dask/ensemble/randomforestclassifier.py b/python/cuml/dask/ensemble/randomforestclassifier.py index c1d3c8ab4e..50c6994167 100755 --- 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