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``.