better use of flush

Author: francesconazzaro

cads_worker/worker.py

@@ -44,14 +44,16 @@ def __init__(self, job_id: str, logger: Any, write_type: str = "stdout"):
 
     def write(self, message: str) -> None:
         """Use the logger as a file-like object. Needed by tqdm progress bar."""
-        if self.write_type == "stdout":
-            self.add_stdout(message)
-        elif self.write_type == "stderr":
-            self.add_stderr(message)
+        self.messages_buffer += message + "\n"
 
     def flush(self) -> None:
         """Write to the logger the content of the buffer."""
-        pass
+        if self.messages_buffer:
+            if self.write_type == "stdout":
+                self.add_stdout(self.messages_buffer)
+            elif self.write_type == "stderr":
+                self.add_stderr(self.messages_buffer)
+            self.messages_buffer = ""
 
     @ensure_session
     def add_user_visible_log(self, message: str, session: Any = None) -> None: