From 477757eac790e89bd1b1ff99debe99d2a58e7412 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 6 Jul 2026 10:42:52 +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/diffusion_dit_trainer.py | 1 -
 studio/backend/tests/test_diffusion_dit_trainer.py    | 6 +++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/studio/backend/core/training/diffusion_dit_trainer.py b/studio/backend/core/training/diffusion_dit_trainer.py
index b6c5cd6983..325bf0d95e 100644
--- a/studio/backend/core/training/diffusion_dit_trainer.py
+++ b/studio/backend/core/training/diffusion_dit_trainer.py
@@ -430,7 +430,6 @@ def _resolve_base_precision(cfg, spec, device) -> str:
         if mode == "mxfp8" and device == "cuda":
             try:
                 import torch
-
                 blackwell = torch.cuda.get_device_capability() >= (10, 0)
             except Exception:  # noqa: BLE001 -- probe failure -> treat as unsupported, fail fast
                 blackwell = False
diff --git a/studio/backend/tests/test_diffusion_dit_trainer.py b/studio/backend/tests/test_diffusion_dit_trainer.py
index aaad553107..0e8e8a0370 100644
--- a/studio/backend/tests/test_diffusion_dit_trainer.py
+++ b/studio/backend/tests/test_diffusion_dit_trainer.py
@@ -143,7 +143,11 @@ def test_family_train_infos_sdxl_supports_compile_without_precision_modes(monkey
 
 
 # ── mxfp8 base precision (DiT dense speed mode) ───────────────────────────────
-def _linear(in_features, out_features, bias = False):
+def _linear(
+    in_features,
+    out_features,
+    bias = False,
+):
     import torch.nn as nn
     return nn.Linear(in_features, out_features, bias = bias)