xx

Author: wangxiaoxin-sherie

tests/ut/ops/test_ascend_forwad_context.py

@@ -1,5 +1,5 @@
-import unittest
 import os
+import unittest
 from unittest import mock
 
 from vllm_ascend.ascend_forward_context import get_dispatcher_name
@@ -16,6 +16,7 @@ def test_get_dispatcher_name(self):
         assert result == "TokenDispatcherWithAll2AllV"
         result = get_dispatcher_name(16, False)
         assert result == "TokenDispatcherWithMC2"
-        with mock.patch.dict(os.environ, {"VLLM_ENABLE_FUSED_EXPERTS_ALLGATHER_EP": "1"}):
+        with mock.patch.dict(os.environ,
+                             {"VLLM_ENABLE_FUSED_EXPERTS_ALLGATHER_EP": "1"}):
             result = get_dispatcher_name(16, False)
-            assert result == "TokenDispatcherWithAllGather"
+            assert result == "TokenDispatcherWithAllGather"