v0.0.14 n_jobs==1 fix

Author: CameleoGrey

examples/object_oriented/nqueens/score/NQueensScoreCalculator.py

@@ -27,7 +27,6 @@ def __init__(self):
     def all_different(self, planning_entity_dfs, problem_fact_dfs):
 
         queens_df = planning_entity_dfs["queens"]
-        #print(queens_df)
         same_row_id_counts = (
             queens_df
             .lazy()

greyjack/greyjack/Solver.py

@@ -194,6 +194,7 @@ def _setup_agents(self):
             agents[i].initial_solution = deepcopy(self.initial_solution)
             agents[i].score_precision = deepcopy(self.score_precision)
             agents[i].logging_level = deepcopy(self.logging_level)
+            agents[i].total_agents_count = self.n_jobs
             self.agent_statuses[str(i)] = "alive"
 
         for i in range(self.n_jobs):

greyjack/greyjack/agents/base/Agent.py

@@ -39,6 +39,7 @@ def __init__(self, migration_rate, migration_frequency, termination_strategy):
         self.steps_to_send_updates = migration_frequency
         self.agent_status = "alive"
         self.round_robin_status_dict = {}
+        self.total_agents_count = None
 
         # windows updates send/receive
         self.context = None
@@ -141,9 +142,10 @@ def solve(self):
                 if self.logging_level in ["trace"] and self.agent_status == "alive":
                     self.logger.info(f"Agent: {self.agent_id} Step: {step_id} Global best: {self.agent_top_individual.score}, Step time: {total_time:.6f}")
 
-                self.steps_to_send_updates -= 1
-                if self.steps_to_send_updates <= 0:
-                    self.send_receive_updates()
+                if self.total_agents_count > 1:
+                    self.steps_to_send_updates -= 1
+                    if self.steps_to_send_updates <= 0:
+                        self.send_receive_updates()
 
                 if self.termination_strategy.is_accomplish():
                     self.agent_status = "dead"