diff --git a/studio/backend/core/training/worker.py b/studio/backend/core/training/worker.py
index df9df699bb..6b526707ce 100644
--- a/studio/backend/core/training/worker.py
+++ b/studio/backend/core/training/worker.py
@@ -1091,6 +1091,7 @@ def run_training_process(
     # AttributeError. Force a full import and stub any missing callables.
     try:
         import torch.distributed as _td
+
         _td_stubs = {
             "is_initialized": lambda: False,
             "is_available": lambda: False,