From bc0d10f7597a7aa7fa56fba4732fe99edfe36107 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Fri, 3 Jul 2026 11:14:46 +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_lora_trainer.py     | 4 +---
 studio/backend/core/training/diffusion_training_service.py | 6 +++++-
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/studio/backend/core/training/diffusion_lora_trainer.py b/studio/backend/core/training/diffusion_lora_trainer.py
index 10b6cd89cd..84bbb6bf74 100644
--- a/studio/backend/core/training/diffusion_lora_trainer.py
+++ b/studio/backend/core/training/diffusion_lora_trainer.py
@@ -512,9 +512,7 @@ def run_diffusion_lora_training(
             grad_norm = None
             if cfg.max_grad_norm and cfg.max_grad_norm > 0:
                 # The returned value is the total PRE-clip norm, reported to the UI chart.
-                grad_norm = float(
-                    torch.nn.utils.clip_grad_norm_(lora_params, cfg.max_grad_norm)
-                )
+                grad_norm = float(torch.nn.utils.clip_grad_norm_(lora_params, cfg.max_grad_norm))
             optimizer.step()
             lr_sched.step()
 
diff --git a/studio/backend/core/training/diffusion_training_service.py b/studio/backend/core/training/diffusion_training_service.py
index 2251038b78..9fbf9fdf25 100644
--- a/studio/backend/core/training/diffusion_training_service.py
+++ b/studio/backend/core/training/diffusion_training_service.py
@@ -88,7 +88,11 @@ def _idle_state() -> dict[str, Any]:
 
 
 def _append_metric(
-    state: dict[str, Any], step: Any, loss: Any, lr: Any, grad_norm: Any = None
+    state: dict[str, Any],
+    step: Any,
+    loss: Any,
+    lr: Any,
+    grad_norm: Any = None,
 ) -> None:
     """Append one (step, loss, lr, grad_norm) point to the bounded history arrays on
     ``state``.