disable yapf for import (conflict with isort)

Signed-off-by: Han Qi <hanq@google.com>

Author: qihqi

tests/layers/vllm/test_compressed_tensors_moe.py

@@ -15,16 +15,18 @@
                                              init_distributed_environment)
 from vllm.engine.arg_utils import EngineArgs
 from vllm.model_executor.layers.fused_moe import FusedMoE
-from vllm.model_executor.layers.fused_moe.config import (FusedMoEConfig,
-                                                         FusedMoEParallelConfig
-                                                         )
+# yapf: disable
+from vllm.model_executor.layers.fused_moe.config import (
+    FusedMoEConfig, FusedMoEParallelConfig)
 
 from tpu_inference.layers.vllm.quantization import get_tpu_quantization_config
 from tpu_inference.layers.vllm.quantization.compressed_tensors.compressed_tensors import \
     VllmCompressedTensorsConfig
 from tpu_inference.layers.vllm.quantization.compressed_tensors.compressed_tensors_moe import \
     VllmCompressedTensorsW8A8Fp8MoEMethod
 
+# yapf: enable
+
 P = PartitionSpec
 
 os.environ['VLLM_DISABLE_SHARED_EXPERTS_STREAM'] = '1'