fix lint

.github/workflows/ruff.yml

@@ -31,4 +31,4 @@ jobs:
         ruff vllm tests
     - name: Spelling check with codespell
       run: |
-         codespell --toml pyproject.toml
\ No newline at end of file
+        codespell --toml pyproject.toml
\ No newline at end of file

tests/kernels/test_prefix_prefill.py

@@ -36,8 +36,8 @@ def test_contexted_kv_attention(
         torch.cuda.manual_seed(0)
     torch.set_default_device(device)
 
-    # Need this, otherwise when we capture the graph the process for GPU 1 would run on both
-    # GPU0 and GPU1 and things would hang
+    # Need this, otherwise when we capture the graph the process for GPU 1 would
+    # run on both GPU0 and GPU1 and things would hang
     #
     # see also similar issue: https://github.com/Dao-AILab/flash-attention/issues/523
     torch.cuda.set_device(device)