From e6d775d0fc80e677292bb29ab856245f5e27fc39 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 05:08:39 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- studio/backend/core/training/diffusion_train_common.py | 4 +++- .../backend/core/training/diffusion_training_service.py | 8 +++++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/studio/backend/core/training/diffusion_train_common.py b/studio/backend/core/training/diffusion_train_common.py index ef63735185..62a16eef8c 100644 --- a/studio/backend/core/training/diffusion_train_common.py +++ b/studio/backend/core/training/diffusion_train_common.py @@ -341,7 +341,9 @@ def discover_image_caption_pairs( # 2. metadata row keyed by file name (basename or the relative path; as_posix so a # Windows backslash path still matches the jsonl's forward-slash keys). if caption is None: - caption = meta_caption.get(img.name) or meta_caption.get(img.relative_to(root).as_posix()) + caption = meta_caption.get(img.name) or meta_caption.get( + img.relative_to(root).as_posix() + ) # 3. dreambooth instance prompt. if caption is None and instance_prompt: caption = instance_prompt diff --git a/studio/backend/core/training/diffusion_training_service.py b/studio/backend/core/training/diffusion_training_service.py index f53fd4623e..5c6a184a86 100644 --- a/studio/backend/core/training/diffusion_training_service.py +++ b/studio/backend/core/training/diffusion_training_service.py @@ -87,7 +87,13 @@ def _idle_state() -> dict[str, Any]: } -def _append_metric(state: dict[str, Any], step: Any, loss: Any, lr: Any, grad_norm: Any = None) -> None: +def _append_metric( + state: dict[str, Any], + step: Any, + loss: Any, + lr: Any, + grad_norm: Any = None, +) -> None: """Append one (step, loss, lr, grad_norm) point to the bounded history arrays on ``state``.