Skip to content

Commit cfb67ae

Browse files
committed
Merge remote-tracking branch 'origin/custom_diff_evo' into constrained_optimization_tunable
2 parents 28a149e + 2d41660 commit cfb67ae

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

kernel_tuner/strategies/diff_evo.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def indices_to_values(individual_indices, tune_params):
7070
values = []
7171
for dim, idx in enumerate(individual_indices):
7272
values.append(tune_params_list[dim][idx])
73-
return np.array(values)
73+
return values
7474

7575

7676
def parse_method(method):
@@ -103,15 +103,15 @@ def generate_population(tune_params, max_idx, popsize, searchspace, constraint_a
103103
if constraint_aware:
104104
samples = LatinHypercube(len(tune_params)).integers(l_bounds=0, u_bounds=max_idx, n=popsize, endpoint=True)
105105
population = [indices_to_values(sample, tune_params) for sample in samples]
106-
population = np.array([repair(individual, searchspace) for individual in population])
106+
population = [repair(individual, searchspace) for individual in population]
107107
else:
108108
population = []
109109
for _ in range(popsize):
110110
ind = []
111111
for key in tune_params:
112112
ind.append(random.choice(tune_params[key]))
113113
population.append(ind)
114-
population = np.array(population)
114+
population = population
115115
return population
116116

117117

@@ -353,7 +353,7 @@ def binomial_crossover(donor_vector, target, CR):
353353
# Apply crossover
354354
trial_vector[crossover_points] = donor_vector[crossover_points]
355355

356-
return trial_vector
356+
return list(trial_vector)
357357

358358

359359
def exponential_crossover(donor_vector, target, CR):
@@ -378,7 +378,7 @@ def exponential_crossover(donor_vector, target, CR):
378378
trial_idx[crossover_point] = donor_vector[crossover_point]
379379
l += 1
380380

381-
return trial_idx
381+
return list(trial_idx)
382382

383383

384384
def repair(trial_vector, searchspace):
@@ -393,7 +393,7 @@ def repair(trial_vector, searchspace):
393393

394394
# if we have found valid neighboring configurations, select one at random
395395
if len(neighbors) > 0:
396-
new_trial_vector = np.array(list(random.choice(neighbors)))
396+
new_trial_vector = list(random.choice(neighbors))
397397
print(f"Differential evolution resulted in invalid config {trial_vector=}, repaired to {new_trial_vector=}")
398398
return new_trial_vector
399399

0 commit comments

Comments
 (0)