fixed bug with imap chunksize

f0uriest · Jan 1, 2019 · 86c2c14 · 86c2c14
1 parent 8217c75
commit 86c2c14
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/gastop/genalg.py b/gastop/genalg.py
@@ -197,7 +197,7 @@ def score(truss):
         # With parallelization
         else:
             if self.ga_params['pop_size'] < 1e4:
-                chunksize = self.ga_params['pop_size']/100
+                chunksize = int(self.ga_params['pop_size']/100)
             else:
                 chunksize = int(np.sqrt(self.ga_params['pop_size']))
             for current_gen in tqdm(range(num_generations), desc='Overall', position=0):
@@ -347,6 +347,7 @@ def update_population(self):  # Cristian
 
         # Append separate lists to form new generation
         population = pop_elite + pop_crossover + pop_mutation + pop_random
-
+        pbar.close()
         # Update population attribute
         self.population = population
+        return