From 11a00d0238c854ffff673a26828a4bf38f64b526 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 14:55:53 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 studio/backend/tests/test_diffusion_arch_patches.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/studio/backend/tests/test_diffusion_arch_patches.py b/studio/backend/tests/test_diffusion_arch_patches.py
index 02e975c5a8..ceeb047669 100644
--- a/studio/backend/tests/test_diffusion_arch_patches.py
+++ b/studio/backend/tests/test_diffusion_arch_patches.py
@@ -245,7 +245,10 @@ def test_krea2_forward_matches_stock():
 
     torch.manual_seed(4)
     blk = Krea2TransformerBlock(
-        hidden_size = D, intermediate_size = 2 * D, num_heads = H, num_kv_heads = H // 2,
+        hidden_size = D,
+        intermediate_size = 2 * D,
+        num_heads = H,
+        num_kv_heads = H // 2,
         norm_eps = 1e-6,
     ).eval()
     # Give the zero-init modulation table real values so all six scale/shift/gate