From b799b240ffdfb3cc408afce8986beec1e4069b02 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sun, 26 Jul 2026 21:43:38 +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_dit_trainer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/studio/backend/tests/test_diffusion_dit_trainer.py b/studio/backend/tests/test_diffusion_dit_trainer.py index ba724b28d3..c64c9065d9 100644 --- a/studio/backend/tests/test_diffusion_dit_trainer.py +++ b/studio/backend/tests/test_diffusion_dit_trainer.py @@ -213,7 +213,9 @@ def test_family_train_infos_lists_dit_families(dit_train_host): assert "4bit" in infos["z-image"]["default_base"].lower() -def test_family_train_infos_sdxl_supports_compile_without_precision_modes(monkeypatch, dit_train_host): +def test_family_train_infos_sdxl_supports_compile_without_precision_modes( + monkeypatch, dit_train_host +): # Regional compile now applies to every family (the SDXL trainer compiles its U-Net blocks too), # but base_precision stays DiT-only, so SDXL advertises no precision modes while z-image keeps # its own. Pin the precision list so the assertion holds regardless of the host GPU.