diff --git a/src/poli_baselines/solvers/multi_objective/nsga_ii.py b/src/poli_baselines/solvers/multi_objective/nsga_ii.py index d41dfc6..b45c8a2 100644 --- a/src/poli_baselines/solvers/multi_objective/nsga_ii.py +++ b/src/poli_baselines/solvers/multi_objective/nsga_ii.py @@ -210,8 +210,8 @@ def __init__( y0_for_init = np.vstack([self.y0, missing_evaluations]) elif self.x0.shape[0] > population_size: best_performing_indices = np.argsort(self.y0.flatten())[::-1] - x0_for_init = self.x0[best_performing_indices[: self.pop_size]] - y0_for_init = y0[best_performing_indices[: self.pop_size]] + x0_for_init = self.x0[best_performing_indices[:population_size]] + y0_for_init = y0[best_performing_indices[:population_size]] else: x0_for_init = self.x0