fix format

Signed-off-by: wangli <wangli858794774@gmail.com>

Author: Potabk

tests/ut/models/test_deepseek_v2.py

@@ -242,9 +242,9 @@ def test_custom_deepseek_v2_mla_attention(mock_rms_norm, mock_distributed,
     positions = torch.arange(4).repeat(2, 1)
     with patch.object(attn.mla_attn,
                       "__call__",
-                      return_value=torch.randn(2, 4, 128)):
-        with pytest.raises(AssertionError):
-            attn(positions, x)
+                      return_value=torch.randn(
+                          2, 4, 128)), pytest.raises(AssertionError):
+        attn(positions, x)
 
     attn = CustomDeepseekV2MLAAttention(config=base_config,
                                         hidden_size=128,