Address lint

Signed-off-by: Yikun Jiang <yikunkero@gmail.com>

Author: Yikun

tests/singlecard/ops/test_rotary_embedding.py

@@ -198,6 +198,7 @@ def test_rotary_embedding_quant_with_leading_dim(
                                atol=DEFAULT_ATOL,
                                rtol=DEFAULT_RTOL)
 
+
 # test rope with npu_mrope interface with leading dimension and merge seqlen and batch_size as num_tokens
 @pytest.mark.parametrize("is_neox_style", IS_NEOX_STYLE)
 @pytest.mark.parametrize("batch_size", BATCH_SIZES)
@@ -244,14 +245,13 @@ def test_npu_mrope_quant_with_leading_dim(
     ref_query, ref_key = rope.forward_native(positions, query, key)
 
     query, key = torch_npu.npu_mrope(
-            positions,
-            query,
-            key,
-            rope.cos_sin_cache,
-            rope.head_size,
-            mrope_section=[0,0,0],
-            rotary_mode='half' if rope.is_neox_style else 'interleave'
-        )
+        positions,
+        query,
+        key,
+        rope.cos_sin_cache,
+        rope.head_size,
+        mrope_section=[0, 0, 0],
+        rotary_mode='half' if rope.is_neox_style else 'interleave')
 
     # Compare the results.
     torch.testing.assert_close(query.view(ref_query.size()),
@@ -261,4 +261,4 @@ def test_npu_mrope_quant_with_leading_dim(
     torch.testing.assert_close(key.view(ref_key.size()),
                                ref_key,
                                atol=DEFAULT_ATOL,
-                               rtol=DEFAULT_RTOL)
+                               rtol=DEFAULT_RTOL)

vllm_ascend/ops/rotary_embedding.py

@@ -70,9 +70,8 @@ def rope_forward_oot(
             key,
             self.cos_sin_cache,
             self.head_size,
-            mrope_section=[0,0,0],
-            rotary_mode='half' if neox_style else 'interleave'
-        )
+            mrope_section=[0, 0, 0],
+            rotary_mode='half' if neox_style else 'interleave')
     return query.view(query_shape), key.view(key_shape)