diff --git a/python/cuml/cuml/cluster/hdbscan/hdbscan.pyx b/python/cuml/cuml/cluster/hdbscan/hdbscan.pyx index dcc5fefaeb..20cfdc397a 100644 --- a/python/cuml/cuml/cluster/hdbscan/hdbscan.pyx +++ b/python/cuml/cuml/cluster/hdbscan/hdbscan.pyx @@ -771,7 +771,7 @@ class HDBSCAN(UniversalBase, ClusterMixin, CMajorInputTagMixin): """ Fit HDBSCAN model from features. """ - + self._all_finite = True X_m, n_rows, n_cols, self.dtype = \ input_to_cuml_array(X, order='C', check_dtype=[np.float32], @@ -1163,7 +1163,7 @@ class HDBSCAN(UniversalBase, ClusterMixin, CMajorInputTagMixin): def get_attr_names(self): attr_names = ['labels_', 'probabilities_', 'cluster_persistence_', 'condensed_tree_', 'single_linkage_tree_', - 'outlier_scores_'] + 'outlier_scores_', '_all_finite'] if self.gen_min_span_tree: attr_names = attr_names + ['minimum_spanning_tree_'] if self.prediction_data: diff --git a/python/cuml/cuml/experimental/accel/estimator_proxy.py b/python/cuml/cuml/experimental/accel/estimator_proxy.py index 7da5cdefb7..429a600dd4 100644 --- a/python/cuml/cuml/experimental/accel/estimator_proxy.py +++ b/python/cuml/cuml/experimental/accel/estimator_proxy.py @@ -23,6 +23,7 @@ from cuml.internals import logger from cuml.internals.global_settings import GlobalSettings from cuml.internals.safe_imports import gpu_only_import, cpu_only_import +from sklearn.base import _clone_parametrized from typing import Optional, Tuple, Dict, Any, Type, List @@ -277,6 +278,9 @@ def __reduce__(self): ), ) + def __sklearn_clone__(self): + return _clone_parametrized(self) + logger.debug( f"Created proxy estimator: ({module_b}, {original_class_name}, {ProxyEstimator})" )