Fix distributed logging [release]

Author: AjayP13

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "DataDreamer"
-version = "0.40.0"
+version = "0.41.0"
 description = "Prompt. Generate Synthetic Data. Train & Align Models."
 license = "MIT"
 authors= [

src/utils/background_utils.py

@@ -170,8 +170,11 @@ def format_for_worker_logging(_fmt: str) -> str:
         lgr.__dict__.update(log_dict)
         for level, formatter in log_handler:
             if "DATADREAMER_DISTRIBUTED" in env:
-                formatter._style._fmt = format_for_worker_logging(formatter._style._fmt)
-                formatter._fmt = format_for_worker_logging(formatter._fmt)
+                if formatter is not None:
+                    formatter._style._fmt = format_for_worker_logging(
+                        formatter._style._fmt
+                    )
+                    formatter._fmt = format_for_worker_logging(formatter._fmt)
             stderr_handler = StreamHandler()
             stderr_handler.setLevel(level)
             stderr_handler.setFormatter(formatter)