diff --git a/scripts/train/train.py b/scripts/train/train.py
index 004f646430..c3da1f1d3c 100644
--- a/scripts/train/train.py
+++ b/scripts/train/train.py
@@ -438,8 +438,10 @@ def main(cfg: DictConfig) -> Trainer:
             format=
             f'%(asctime)s: rank{dist.get_global_rank()}[%(process)d][%(threadName)s]: %(levelname)s: %(name)s: %(message)s'
         )
-        logging.getLogger('llmfoundry').setLevel(python_log_level.upper())  # Foundry module
-        logging.getLogger(__name__).setLevel(python_log_level.upper())  # Train script
+        logging.getLogger('llmfoundry').setLevel(
+            python_log_level.upper())  # Foundry module
+        logging.getLogger(__name__).setLevel(
+            python_log_level.upper())  # Train script
 
     # Initialize context
     init_context = process_init_device(model_config, fsdp_config)