type fix

rapidsai · Aug 22, 2024 · 337aa88 · 337aa88
1 parent 7dbe38d
commit 337aa88
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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):