diff --git a/diversity_algorithms/algorithms/quality_diversity.py b/diversity_algorithms/algorithms/quality_diversity.py index 1b7dab1..a612477 100644 --- a/diversity_algorithms/algorithms/quality_diversity.py +++ b/diversity_algorithms/algorithms/quality_diversity.py @@ -110,7 +110,7 @@ def get_nov(self, bd, extra_indivs=[], in_archive=False): for ind in extra_indivs: dists.append(np.linalg.norm(np.array(bd)-np.array(ind.bd))) # Query KNN in archive - dists_archive, _ = self.kdtree.query(np.array(bd),self.k+1, n_jobs=-1) + dists_archive, _ = self.kdtree.query(np.array(bd),self.k+1, workers=-1) dists += list(dists_archive) dists.sort() if(in_archive): @@ -204,7 +204,7 @@ def get_nov(self, bd, extra_indivs=[], in_archive=False): for ind in extra_indivs: dists.append(np.linalg.norm(np.array(bd)-np.array(ind.bd))) # Query KNN in archive - dists_archive, _ = self.kdtree.query(np.array(bd),self.k+1, n_jobs=-1) + dists_archive, _ = self.kdtree.query(np.array(bd),self.k+1, workers=-1) dists += list(dists_archive) dists.sort() if(in_archive): @@ -223,7 +223,7 @@ def post_add_update(self): def try_add(self,indiv): bd = indiv.bd - close_neighbors = ([] if((self.kdtree is None) or (self.r == 0)) else self.kdtree.query_ball_point(bd, self.r, n_jobs=-1)) + close_neighbors = ([] if((self.kdtree is None) or (self.r == 0)) else self.kdtree.query_ball_point(bd, self.r, workers=-1)) if not close_neighbors: # No neighbors in ball, no problem - add indiv self.archive.append(indiv) self.post_add_update()