Fix bug indies of wl and wg variables

thieu1995 · Jan 3, 2024 · b628476 · b628476
1 parent 5117303
commit b628476
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mealpy/human_based/WarSO.py b/mealpy/human_based/WarSO.py
@@ -69,7 +69,9 @@ def evolve(self, epoch):
         Args:
             epoch (int): The current iteration
         """
-        pop_sorted = self.get_sorted_population(self.pop, self.problem.minmax)
+        pop_sorted, indices = self.get_sorted_population(self.pop, self.problem.minmax, return_index=True)
+        self.wl = self.wl[indices]
+        self.wg = self.wg[indices]
         com = self.generator.permutation(self.pop_size)
         for idx in range(0, self.pop_size):
             r1 = self.generator.random()