From 7f237ca57edc863579ae793d9154b8bb7f80fea7 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 20 Apr 2026 16:17:30 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

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

diff --git a/scripts/benchmarks/qwen3_flex_inference.py b/scripts/benchmarks/qwen3_flex_inference.py
index e178c9f404..6b9bc7737b 100644
--- a/scripts/benchmarks/qwen3_flex_inference.py
+++ b/scripts/benchmarks/qwen3_flex_inference.py
@@ -58,7 +58,9 @@ from flex_paged_attention import PagedKVCache, PageTable  # noqa: E402
 _FLEX_COMPILE_MODE = os.environ.get("FLEX_COMPILE_MODE", None)
 if _FLEX_COMPILE_MODE:
     flex_attention_compiled = torch.compile(
-        flex_attention, fullgraph = True, mode = _FLEX_COMPILE_MODE,
+        flex_attention,
+        fullgraph = True,
+        mode = _FLEX_COMPILE_MODE,
     )
 else:
     flex_attention_compiled = torch.compile(flex_attention, fullgraph = True)