From 5f7d99b4bded0b116eaafdc06cadcd6ce72fff5a Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Wed, 10 Jun 2026 06:57:10 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 unsloth/kernels/flex_attention.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/unsloth/kernels/flex_attention.py b/unsloth/kernels/flex_attention.py
index 512e6361ff..7e9c80cf13 100644
--- a/unsloth/kernels/flex_attention.py
+++ b/unsloth/kernels/flex_attention.py
@@ -47,7 +47,9 @@ def _flex_is_dgx_spark():
 
         out = subprocess.run(
             ["nvidia-smi", "--query-gpu=name", "--format=csv,noheader"],
-            capture_output = True, text = True, timeout = 5,
+            capture_output = True,
+            text = True,
+            timeout = 5,
         )
         names = (out.stdout or "").upper()
         return any(t in names for t in ("GB10", "JMJWOA", "N1X", "DGX SPARK", "GB110"))