diff --git a/studio/backend/tests/test_diffusion_lora_trainer.py b/studio/backend/tests/test_diffusion_lora_trainer.py index e238fdaff7..fd17b29500 100644 --- a/studio/backend/tests/test_diffusion_lora_trainer.py +++ b/studio/backend/tests/test_diffusion_lora_trainer.py @@ -208,9 +208,7 @@ def test_config_rejects_known_non_sdxl_base_models(): "z-image-turbo-Q4_K_M.gguf", ): with pytest.raises(ValueError, match = "SDXL"): - DiffusionLoraConfig( - base_model = bad, data_dir = "d", output_dir = "o" - ).normalized() + DiffusionLoraConfig(base_model = bad, data_dir = "d", output_dir = "o").normalized() def test_config_accepts_sdxl_and_unknown_base_models():