From 4cbb7ceecd6210da657043abe03d985ec44c2333 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 1 Jul 2026 05:08:40 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 studio/backend/core/inference/diffusion.py    |  8 +++--
 .../tests/test_diffusion_controlnet.py        | 35 +++++++++++++++----
 2 files changed, 33 insertions(+), 10 deletions(-)

diff --git a/studio/backend/core/inference/diffusion.py b/studio/backend/core/inference/diffusion.py
index 6bbe30b0b0..632b526ca6 100644
--- a/studio/backend/core/inference/diffusion.py
+++ b/studio/backend/core/inference/diffusion.py
@@ -1156,9 +1156,11 @@ class DiffusionBackend:
         if cn_model is None:
             if cancel.is_set():
                 raise RuntimeError("Diffusion generation was cancelled.")
-            cn_model = getattr(diffusers, model_cls_name).from_pretrained(
-                resolved_cn.path, torch_dtype = state.dtype, token = state.hf_token
-            ).to(state.device)
+            cn_model = (
+                getattr(diffusers, model_cls_name)
+                .from_pretrained(resolved_cn.path, torch_dtype = state.dtype, token = state.hf_token)
+                .to(state.device)
+            )
             self._cn_models[resolved_cn.id] = cn_model
         key = (pipe_cls_name, resolved_cn.id)
         pipe = self._cn_pipes.get(key)
diff --git a/studio/backend/tests/test_diffusion_controlnet.py b/studio/backend/tests/test_diffusion_controlnet.py
index fb267ca71f..9c978a483c 100644
--- a/studio/backend/tests/test_diffusion_controlnet.py
+++ b/studio/backend/tests/test_diffusion_controlnet.py
@@ -78,12 +78,18 @@ def test_supports_controlnet_matrix():
     assert not dc.supports_controlnet(**ok, model_kind = "single_file", transformer_quant = "int8")
     # native engine + a family without a CN pipeline are off.
     assert not dc.supports_controlnet(
-        engine = "sd_cpp", family = "flux.1", has_controlnet_pipeline = True,
-        model_kind = "gguf", transformer_quant = None,
+        engine = "sd_cpp",
+        family = "flux.1",
+        has_controlnet_pipeline = True,
+        model_kind = "gguf",
+        transformer_quant = None,
     )
     assert not dc.supports_controlnet(
-        engine = "diffusers", family = "z-image", has_controlnet_pipeline = False,
-        model_kind = "pipeline", transformer_quant = None,
+        engine = "diffusers",
+        family = "z-image",
+        has_controlnet_pipeline = False,
+        model_kind = "pipeline",
+        transformer_quant = None,
     )
 
 
@@ -96,7 +102,12 @@ def test_controlnet_spec_and_request_validation():
     assert DiffusionGenerateRequest(prompt = "x").controlnet is None
     req = DiffusionGenerateRequest(
         prompt = "x",
-        controlnet = {"id": "flux-union-pro", "image": "data", "control_type": "canny", "strength": 0.6},
+        controlnet = {
+            "id": "flux-union-pro",
+            "image": "data",
+            "control_type": "canny",
+            "strength": 0.6,
+        },
     )
     assert req.controlnet.id == "flux-union-pro" and req.controlnet.strength == 0.6
     # defaults
@@ -129,7 +140,12 @@ def test_families_declare_controlnet_classes():
 
 class _FakeCNModel:
     @classmethod
-    def from_pretrained(cls, path, torch_dtype = None, token = None):
+    def from_pretrained(
+        cls,
+        path,
+        torch_dtype = None,
+        token = None,
+    ):
         m = cls()
         m.path = path
         return m
@@ -141,7 +157,12 @@ class _FakeCNModel:
 
 class _FakeCNPipe:
     @classmethod
-    def from_pipe(cls, base, controlnet = None, torch_dtype = None):
+    def from_pipe(
+        cls,
+        base,
+        controlnet = None,
+        torch_dtype = None,
+    ):
         p = cls()
         p.base = base
         p.controlnet = controlnet