Fixes bug where number of iterations is printed incorrectly (#10)

SneaksAndData · Jul 7, 2022 · 663890f · 663890f
1 parent a308872
commit 663890f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/anti_clustering/simulated_annealing_heuristic.py b/anti_clustering/simulated_annealing_heuristic.py
@@ -51,7 +51,7 @@ def _solve(self, distance_matrix: npt.NDArray[float], num_groups: int) -> npt.ND
         objective = self._calculate_objective(cluster_assignment, distance_matrix)
         for iteration in range(self.iterations):
             if self.verbose and iteration % 5 == 0:
-                print(f'Iteration {iteration + 1} of {len(distance_matrix)}')
+                print(f'Iteration {iteration + 1} of {self.iterations}')
 
             # Select random element
             i = self.rnd.randint(0, len(distance_matrix) - 1)