From 4af1200dd3a90ac5990b72fbef3cfcf51589c4e1 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Fri, 12 Jun 2026 05:22:45 +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/worker.py | 2 ++
 unsloth/kernels/flex_attention.py      | 1 +
 2 files changed, 3 insertions(+)

diff --git a/studio/backend/core/training/worker.py b/studio/backend/core/training/worker.py
index f52dfc48f1..77628e4a44 100644
--- a/studio/backend/core/training/worker.py
+++ b/studio/backend/core/training/worker.py
@@ -750,6 +750,7 @@ def _nvidia_classify_spark_unified_memory(props: Any) -> tuple[str, bool]:
         return "is_integrated", True
     name_upper = (getattr(props, "name", "") or "").upper()
     import re
+
     for token in ("GB10", "GB110", "JMJWOA", "N1X", "DGX SPARK"):
         # Whole-token match so "GB10" does not match a discrete "GB100"/"GB10X".
         if re.search(r"(?<![A-Z0-9])" + re.escape(token) + r"(?![A-Z0-9])", name_upper):
@@ -2458,6 +2459,7 @@ def run_training_process(*, event_queue: Any, stop_queue: Any, config: dict) ->
                     )
                     _names_u = (_smi.stdout or "").upper()
                     import re as _re
+
                     _spark_smi = any(
                         _re.search(r"(?<![A-Z0-9])" + _re.escape(t) + r"(?![A-Z0-9])", _names_u)
                         for t in ("GB10", "GB110", "JMJWOA", "N1X", "DGX SPARK")
diff --git a/unsloth/kernels/flex_attention.py b/unsloth/kernels/flex_attention.py
index 36bc3a84e8..71428b27b1 100644
--- a/unsloth/kernels/flex_attention.py
+++ b/unsloth/kernels/flex_attention.py
@@ -51,6 +51,7 @@ def _flex_is_dgx_spark():
         names = (out.stdout or "").upper()
         # Whole-token match so "GB10" does not match a discrete "GB100"/"GB10X".
         import re
+
         return any(
             re.search(r"(?<![A-Z0-9])" + re.escape(t) + r"(?![A-Z0-9])", names)
             for t in ("GB10", "JMJWOA", "N1X", "DGX SPARK", "GB110")