fixed comm

Helmholtz-AI-Energy · May 15, 2024 · 8a75b20 · 8a75b20
1 parent a7ed381
commit 8a75b20
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/propulate/pollinator.py b/propulate/pollinator.py
@@ -415,10 +415,10 @@ def _work(self, logging_interval: int = 10, debug: int = 1):
         # Loop over generations.
         # TODO this should probably be refactored, checkpointing can probably be handled in one place
         with h5py.File(
-            self.checkpoint_path, "a", driver="mpio", comm=MPI.COMM_WORLD
+            self.checkpoint_path, "a", driver="mpio", comm=self.propulate_comm
         ) as f:
-            while self.generations <= -1 or self.generation < self.generations:
-                if debug == 1 and self.generation % int(logging_interval) == 0:
+            while self.generation < self.generations:
+                if self.generation % int(logging_interval) == 0:
                     log.info(
                         f"Island {self.island_idx} Worker {self.island_comm.rank}: In generation {self.generation}..."
                     )