From a288ff208cd1d5c68ea3131fbf4f1c3f6604790a Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Fri, 8 May 2026 09:58:25 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 studio/backend/core/training/worker.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/studio/backend/core/training/worker.py b/studio/backend/core/training/worker.py
index 28cecfcdb2..fa334a825e 100644
--- a/studio/backend/core/training/worker.py
+++ b/studio/backend/core/training/worker.py
@@ -1100,11 +1100,13 @@ def run_training_process(
     }
     try:
         import torch.distributed as _td
+
         for _name, _stub in _td_stubs.items():
             if not hasattr(_td, _name):
                 setattr(_td, _name, _stub)
     except Exception:
         import types
+
         _td_mock = types.ModuleType("torch.distributed")
         for _name, _stub in _td_stubs.items():
             setattr(_td_mock, _name, _stub)
@@ -1115,6 +1117,7 @@ def run_training_process(
         )
         try:
             import torch as _torch
+
             _torch.distributed = _td_mock
         except Exception:
             pass