Update when branching levels are done

algo-hhu · Jul 24, 2024 · 47f4023 · 47f4023
1 parent 54d0386
commit 47f4023
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/exact_kmeans/ilp.py b/exact_kmeans/ilp.py
@@ -709,7 +709,7 @@ def enumerate_sizes(
                 # It does not make sense to run it if we only have one value
                 # Because we have done it before with the other ILP
                 if (
-                    len(current_cluster_sizes) > 2
+                    len(current_cluster_sizes) >= 2
                     and len(current_cluster_sizes) <= self.ilp_branching_until_level
                 ):
                     if self.config.get("fill_cluster_sizes", False):