[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
2026-04-20 16:17:30 +00:00 · 2026-04-20 16:17:30 +00:00 · 7f237ca57e
commit 7f237ca57e
parent 816c5fb78d
1 changed files with 3 additions and 1 deletions
--- 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)