isort

Author: gshtras

benchmarks/kernels/benchmark_paged_attention.py

@@ -6,7 +6,7 @@
 
 from vllm import _custom_ops as ops
 from vllm.utils import (STR_DTYPE_TO_TORCH_DTYPE, FlexibleArgumentParser,
-                        create_kv_caches_with_random, seed_everything, is_hip)
+                        create_kv_caches_with_random, is_hip, seed_everything)
 
 NUM_BLOCKS = 1024 * 1024
 PARTITION_SIZE = 512

tests/kernels/test_moe.py

@@ -21,7 +21,7 @@
     marlin_quantize)
 from vllm.model_executor.models.mixtral import MixtralMoE
 from vllm.scalar_type import scalar_types
-from vllm.utils import seed_everything, is_hip
+from vllm.utils import is_hip, seed_everything
 
 
 def torch_moe(a, w1, w2, score, topk):

vllm/config.py

@@ -905,8 +905,8 @@ def _verify_args(self) -> None:
         if self.use_ray:
             from vllm.executor import ray_utils
             ray_utils.assert_ray_available()
-        if (not self.disable_custom_all_reduce and self.world_size > 1 and 
-                self.pipeline_parallel_size > 1):
+        if (not self.disable_custom_all_reduce and self.world_size > 1
+                and self.pipeline_parallel_size > 1):
             self.disable_custom_all_reduce = True
             logger.info(
                 "Disabled the custom all-reduce kernel because it is not "