lint

Signed-off-by: Bill Nell <bnell@redhat.com>

Author: bnellnm

tests/kernels/moe/test_batched_moe.py

@@ -9,7 +9,8 @@
 import triton.language as tl
 
 from tests.kernels.moe.utils import (batched_moe, make_test_weights,
-                                     torch_moe2, triton_moe)
+                                     torch_moe2, triton_moe,
+                                     per_block_cast_to_fp8)
 from tests.kernels.quant_utils import native_w8a8_block_matmul
 from vllm.config import VllmConfig, set_current_vllm_config
 from vllm.model_executor.layers.fused_moe.fused_batched_moe import (

tests/kernels/moe/test_pplx_moe.py

@@ -23,13 +23,15 @@
                                      torch_moe2)
 from tests.pplx_utils import ProcessGroupInfo, parallel_launch
 from vllm.config import VllmConfig, set_current_vllm_config
-from vllm.model_executor.layers.fused_moe import override_config
+from vllm.model_executor.layers.fused_moe import (override_config,
+                                                  FusedMoEConfig,
+                                                  fused_topk,
+                                                  get_default_config,
+                                                  FusedMoEModularKernel,
+                                                  BatchedTritonExperts,
+                                                  FusedMoEModularKernel)
 from vllm.model_executor.layers.fused_moe.fused_batched_moe import (
-    BatchedPrepareAndFinalize, BatchedTritonExperts, NaiveBatchedExperts)
-from vllm.model_executor.layers.fused_moe.fused_moe import (fused_topk,
-                                                            get_default_config)
-from vllm.model_executor.layers.fused_moe.modular_kernel import (
-    FusedMoEModularKernel)
+    BatchedPrepareAndFinalize, NaiveBatchedExperts)
 from vllm.platforms import current_platform
 from vllm.utils import round_up