From 62692b998e743a6348efe35c08bc4fb5662efe32 Mon Sep 17 00:00:00 2001 From: Sourcery AI Date: Fri, 10 Sep 2021 19:07:22 +0000 Subject: [PATCH] 'Refactored by Sourcery' --- clusterization/kmeans/kmeans.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/clusterization/kmeans/kmeans.py b/clusterization/kmeans/kmeans.py index 852844d..7b4ae1c 100644 --- a/clusterization/kmeans/kmeans.py +++ b/clusterization/kmeans/kmeans.py @@ -110,10 +110,7 @@ def _assign(self): cluster_index = None for ci, c in enumerate(self.clusters): distance = p.distance(c.mean) - if min_distance is None: - min_distance = distance - cluster_index = ci - elif distance < min_distance: + if min_distance is None or distance < min_distance: min_distance = distance cluster_index = ci tmp_clusters[cluster_index].append(p)