From 176c9a9f86f4d2cf9c0fd9f12075d33792fd79b6 Mon Sep 17 00:00:00 2001
From: Ben Frederickson <ben@benfrederickson.com>
Date: Tue, 1 Oct 2024 17:16:54 -0700
Subject: [PATCH] use l2expanded distance in kmeans transform

---
 python/cuml/cuml/cluster/kmeans.pyx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/cuml/cuml/cluster/kmeans.pyx b/python/cuml/cuml/cluster/kmeans.pyx
index e60a58fff8..c45b5ee5d1 100644
--- a/python/cuml/cuml/cluster/kmeans.pyx
+++ b/python/cuml/cuml/cluster/kmeans.pyx
@@ -604,7 +604,7 @@ class KMeans(UniversalBase,
             cdef KMeansParams* params = \
                 <KMeansParams*><size_t>self._get_kmeans_params()
 
-            params.metric = CuvsDistanceType.L2SqrtExpanded
+            params.metric = CuvsDistanceType.L2Expanded
 
             int_dtype = np.int32 if self.labels_.dtype == np.int32 else np.int64