From 337aa8802dd4ca0be37405f87da920472e7ca9f5 Mon Sep 17 00:00:00 2001 From: jinsolp Date: Thu, 22 Aug 2024 22:17:16 +0000 Subject: [PATCH] type fix --- python/cuml/cuml/cluster/hdbscan/hdbscan.pyx | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/cuml/cuml/cluster/hdbscan/hdbscan.pyx b/python/cuml/cuml/cluster/hdbscan/hdbscan.pyx index afbfd34a6d..37ef2ded3d 100644 --- a/python/cuml/cuml/cluster/hdbscan/hdbscan.pyx +++ b/python/cuml/cuml/cluster/hdbscan/hdbscan.pyx @@ -54,7 +54,7 @@ IF GPUBUILD == 1: size_t max_iterations, float termination_threshold, bool return_distances, - uint64_t n_clusters, + size_t n_clusters, cdef extern from "cuml/cluster/hdbscan.hpp" namespace "ML::HDBSCAN::Common": @@ -570,6 +570,8 @@ class HDBSCAN(UniversalBase, ClusterMixin, CMajorInputTagMixin): self.prediction_data_ptr = None self._cpu_to_gpu_interop_prepped = False + logger.set_level(verbose) + @property def condensed_tree_(self):