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)