From 003e28730cf052fe33d714ca683a6d7274c4d8a8 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sat, 4 Jul 2026 03:31:53 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .../backend/core/training/diffusion_dit_trainer.py | 1 - studio/backend/routes/models.py | 4 +--- studio/backend/tests/test_diffusion_dataset_api.py | 6 ++++-- studio/backend/tests/test_diffusion_dit_trainer.py | 9 +++++---- studio/backend/tests/test_diffusion_training.py | 6 ++++-- studio/backend/tests/test_local_model_format.py | 13 +++++++++---- 6 files changed, 23 insertions(+), 16 deletions(-) diff --git a/studio/backend/core/training/diffusion_dit_trainer.py b/studio/backend/core/training/diffusion_dit_trainer.py index 105fd0a822..de98350821 100644 --- a/studio/backend/core/training/diffusion_dit_trainer.py +++ b/studio/backend/core/training/diffusion_dit_trainer.py @@ -356,7 +356,6 @@ def _apply_mxfp8_training(transformer, on_event) -> bool: Never fatal: on any failure the run continues in bf16 with a warning.""" try: from torchao.quantization import quantize_ - quantize_( transformer, _mxfp8_training_config(), diff --git a/studio/backend/routes/models.py b/studio/backend/routes/models.py index 54bc490052..d4501517c8 100644 --- a/studio/backend/routes/models.py +++ b/studio/backend/routes/models.py @@ -911,9 +911,7 @@ async def list_local_models( models = collect_local_models(models_root) # Tag each model with its task so the Images picker can filter to diffusion # (GGUF by architecture; local diffusers checkpoints by pipeline / family). - models = [ - m.model_copy(update = {"task": _local_model_task(m)}) for m in models - ] + models = [m.model_copy(update = {"task": _local_model_task(m)}) for m in models] return LocalModelListResponse( models_dir = str(models_root), diff --git a/studio/backend/tests/test_diffusion_dataset_api.py b/studio/backend/tests/test_diffusion_dataset_api.py index e1c2c0a186..b10dd04f4c 100644 --- a/studio/backend/tests/test_diffusion_dataset_api.py +++ b/studio/backend/tests/test_diffusion_dataset_api.py @@ -64,8 +64,10 @@ def test_list_images_caption_precedence(client, ds_root): # a.png -> sidecar (an explicit edit beats the metadata row), b.png -> metadata-only, # c.png -> none. (folder / "metadata.jsonl").write_text( - json.dumps({"file_name": "a.png", "text": "from metadata"}) + "\n" - + json.dumps({"file_name": "b.png", "text": "from metadata"}) + "\n", + json.dumps({"file_name": "a.png", "text": "from metadata"}) + + "\n" + + json.dumps({"file_name": "b.png", "text": "from metadata"}) + + "\n", encoding = "utf-8", ) (folder / "a.txt").write_text("edited sidecar", encoding = "utf-8") diff --git a/studio/backend/tests/test_diffusion_dit_trainer.py b/studio/backend/tests/test_diffusion_dit_trainer.py index 8d46200e23..b58868c701 100644 --- a/studio/backend/tests/test_diffusion_dit_trainer.py +++ b/studio/backend/tests/test_diffusion_dit_trainer.py @@ -67,7 +67,10 @@ def test_select_lora_targets_explicit_override_wins(): base_model = "black-forest-labs/FLUX.1-dev", data_dir = "d", output_dir = "o" ).normalized() assert cfg.lora_target_modules == DEFAULT_LORA_TARGETS - assert _select_lora_targets(cfg.lora_target_modules, _SPECS["flux.1"].lora_targets) == _FLUX_TARGETS + assert ( + _select_lora_targets(cfg.lora_target_modules, _SPECS["flux.1"].lora_targets) + == _FLUX_TARGETS + ) @pytest.mark.parametrize( @@ -215,9 +218,7 @@ def test_mxfp8_training_config_falls_back_to_the_torchao_0_17_api(monkeypatch): calls["recipe"] = recipe return "cfg-0.17" - fake_config = SimpleNamespace( - MXFP8TrainingOpConfig = _OpConfig, MXFP8TrainingRecipe = _Recipe - ) + fake_config = SimpleNamespace(MXFP8TrainingOpConfig = _OpConfig, MXFP8TrainingRecipe = _Recipe) monkeypatch.setitem(sys.modules, "torchao.prototype.mx_formats", None) monkeypatch.setitem( sys.modules, "torchao.prototype.moe_training", SimpleNamespace(config = fake_config) diff --git a/studio/backend/tests/test_diffusion_training.py b/studio/backend/tests/test_diffusion_training.py index d3748eea35..fd28213294 100644 --- a/studio/backend/tests/test_diffusion_training.py +++ b/studio/backend/tests/test_diffusion_training.py @@ -561,8 +561,10 @@ def test_diffusion_info_counts_metadata_captions(client, dataset_roots): (folder / "c.png").write_bytes(b"x") # a.png + b.png via metadata; a.png also has a sidecar (must count once); c.png none. (folder / "metadata.jsonl").write_text( - json.dumps({"file_name": "a.png", "text": "cap a"}) + "\n" - + json.dumps({"file_name": "b.png", "text": "cap b"}) + "\n", + json.dumps({"file_name": "a.png", "text": "cap a"}) + + "\n" + + json.dumps({"file_name": "b.png", "text": "cap b"}) + + "\n", encoding = "utf-8", ) (folder / "a.txt").write_text("edited a", encoding = "utf-8") diff --git a/studio/backend/tests/test_local_model_format.py b/studio/backend/tests/test_local_model_format.py index dd888eb50d..f9184774d6 100644 --- a/studio/backend/tests/test_local_model_format.py +++ b/studio/backend/tests/test_local_model_format.py @@ -121,7 +121,14 @@ def test_scan_models_dir_classifies_root_gguf_with_config(tmp_path): from models.models import LocalModelInfo # noqa: E402 -def _local(path, *, model_format = None, model_id = None, display_name = "m", id = "m"): +def _local( + path, + *, + model_format = None, + model_id = None, + display_name = "m", + id = "m", +): return LocalModelInfo( id = id, display_name = display_name, @@ -147,9 +154,7 @@ def test_local_task_tags_diffusers_by_family_id(tmp_path): d = tmp_path / "flux-checkpoint" _touch(d / "flux1-dev.safetensors") assert ( - models_route._local_model_task( - _local(d, model_id = "black-forest-labs/FLUX.1-dev") - ) + models_route._local_model_task(_local(d, model_id = "black-forest-labs/FLUX.1-dev")) == "text-to-image" )