From d86c4c70e666f10d1ee590d3f2dbd9098036358a 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 14:05:20 +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_backend.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/studio/backend/tests/test_diffusion_backend.py b/studio/backend/tests/test_diffusion_backend.py
index 119c362ea7..12c9d0c658 100644
--- a/studio/backend/tests/test_diffusion_backend.py
+++ b/studio/backend/tests/test_diffusion_backend.py
@@ -2134,11 +2134,15 @@ def test_dense_quant_prefetch_needed_gates(fake_runtime, monkeypatch):
     # memory_mode overrides it) must NOT widen, even though the candidate itself is dense-viable.
     before = len(seen)
     assert (
-        backend._dense_quant_prefetch_needed(fam, {"transformer_quant": "fp8", "memory_mode": "balanced"})
+        backend._dense_quant_prefetch_needed(
+            fam, {"transformer_quant": "fp8", "memory_mode": "balanced"}
+        )
         is False
     )
     assert (
-        backend._dense_quant_prefetch_needed(fam, {"transformer_quant": "fp8", "memory_mode": "low_vram"})
+        backend._dense_quant_prefetch_needed(
+            fam, {"transformer_quant": "fp8", "memory_mode": "low_vram"}
+        )
         is False
     )
     assert (
@@ -2150,7 +2154,9 @@ def test_dense_quant_prefetch_needed_gates(fake_runtime, monkeypatch):
     # An explicit memory_mode still consulting the candidate: fast/auto can flip resident, so they
     # widen when the candidate is dense-viable (memory_mode="fast" does not force offload).
     assert (
-        backend._dense_quant_prefetch_needed(fam, {"transformer_quant": "fp8", "memory_mode": "fast"})
+        backend._dense_quant_prefetch_needed(
+            fam, {"transformer_quant": "fp8", "memory_mode": "fast"}
+        )
         is True
     )
     # A cpu_offload flag is overridden by an explicit resident memory_mode, so it still widens.