Flush the stdout to avoid waiting aiida workchains

Author: bastonero

Modules/aiida_ensemble.py

@@ -4,6 +4,7 @@
 
 from copy import copy, deepcopy
 import time
+import sys
 
 from ase import units
 from cellconstructor.Structure import Structure
@@ -126,6 +127,8 @@ def compute_ensemble( # pylint: disable=arguments-renamed
                     **kwargs
                 )
 
+                sys.stdout.flush()
+
                 if group:
                     group.add_nodes(workchains)
 
@@ -169,6 +172,8 @@ def compute_ensemble( # pylint: disable=arguments-renamed
                 if self.gp_model is not None:
                     self._train_gp()
                     self._write_model()
+                
+                sys.stdout.flush()
 
         # ================ FINALIZE ================ #
         # if self.has_stress:
@@ -231,6 +236,8 @@ def _predict_with_model(
                     self.stress_computed[index] = True
 
                 self.force_computed[index] = True
+            
+            sys.stdout.flush()
 
 
     def _compute_properties(self, atoms: FLARE_Atoms) -> None: