Make ctypedef conditional

rapidsai · Dec 19, 2024 · 4fb5e03 · 4fb5e03
1 parent 15de75d
commit 4fb5e03
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/cuml/cuml/cluster/kmeans.pyx b/python/cuml/cuml/cluster/kmeans.pyx
@@ -38,6 +38,9 @@ IF GPUBUILD == 1:
     from cuml.cluster.kmeans_utils cimport KMeansPlusPlus, Random, Array
     from cuml.cluster cimport kmeans_utils
 
+    # Avoid potential future conflicts with cuml's level enum
+    ctypedef kmeans_utils.level_enum raft_level_enum
+
 from cuml.internals.array import CumlArray
 from cuml.common.array_descriptor import CumlArrayDescriptor
 from cuml.internals.base import UniversalBase
@@ -53,9 +56,6 @@ _openmp_effective_n_threads = safe_import_from(
     "sklearn.utils._openmp_helpers", "_openmp_effective_n_threads", alt=return_false
 )
 
-# Avoid potential future conflicts with cuml's level enum
-ctypedef kmeans_utils.level_enum raft_level_enum
-
 
 class KMeans(UniversalBase,
              ClusterMixin,