fix lint:

Signed-off-by: ganyi <pleaplusone.gy@gmail.com>

Author: ganyi1996ppo

tests/e2e/singlecard/ops/test_rotary_embedding.py

@@ -248,9 +248,12 @@ def forward(
         o = self.o_proj(query)
         return o
 
+
 # The first graph seems will have some accuracy issue when directly run pytest on the ops folder,
 # add a warmup graph replay for workaround
 ACL_GRPAH_FIRST_RUN = True
+
+
 @pytest.mark.parametrize("is_neox_style", IS_NEOX_STYLE)
 @pytest.mark.parametrize("num_tokens", BATCH_SIZES)
 @pytest.mark.parametrize("num_heads", NUM_HEADS)
@@ -330,10 +333,10 @@ def custom_op_checking_backend(gm: torch.fx.GraphModule, example_input):
     static_hidden_states.copy_(random_filled_hidden_states)
 
     aclgraph.replay()
-    aclgraph.replay()
+    global ACL_GRPAH_FIRST_RUN
     if ACL_GRPAH_FIRST_RUN:
         ACL_GRPAH_FIRST_RUN = False
-        return 
+        return
     output_reference = model(static_positions, static_hidden_states)
     torch.testing.assert_close(static_output,
                                output_reference,