string values compatible

Author: benvanwerkhoven

kernel_tuner/strategies/diff_evo.py

@@ -71,7 +71,7 @@ def indices_to_values(individual_indices, tune_params):
     values = []
     for dim, idx in enumerate(individual_indices):
         values.append(tune_params_list[dim][idx])
-    return np.array(values)
+    return values
 
 
 def parse_method(method):
@@ -104,15 +104,15 @@ def generate_population(tune_params, max_idx, popsize, searchspace, constraint_a
     if constraint_aware:
         samples = LatinHypercube(len(tune_params)).integers(l_bounds=0, u_bounds=max_idx, n=popsize, endpoint=True)
         population = [indices_to_values(sample, tune_params) for sample in samples]
-        population = np.array([repair(individual, searchspace) for individual in population])
+        population = [repair(individual, searchspace) for individual in population]
     else:
         population = []
         for _ in range(popsize):
             ind = []
             for key in tune_params:
                 ind.append(random.choice(tune_params[key]))
             population.append(ind)
-        population = np.array(population)
+        population = population
     return population
 
 
@@ -354,7 +354,7 @@ def binomial_crossover(donor_vector, target, CR):
     # Apply crossover
     trial_vector[crossover_points] = donor_vector[crossover_points]
 
-    return trial_vector
+    return list(trial_vector)
 
 
 def exponential_crossover(donor_vector, target, CR):
@@ -379,7 +379,7 @@ def exponential_crossover(donor_vector, target, CR):
         trial_idx[crossover_point] = donor_vector[crossover_point]
         l += 1
 
-    return trial_idx
+    return list(trial_idx)
 
 
 def repair(trial_vector, searchspace):
@@ -394,7 +394,7 @@ def repair(trial_vector, searchspace):
 
             # if we have found valid neighboring configurations, select one at random
             if len(neighbors) > 0:
-                new_trial_vector = np.array(list(random.choice(neighbors)))
+                new_trial_vector = list(random.choice(neighbors))
                 print(f"Differential evolution resulted in invalid config {trial_vector=}, repaired to {new_trial_vector=}")
                 return new_trial_vector