diff --git a/tests/kernels/moe/modular_kernel_tools/parallel_utils.py b/tests/kernels/moe/modular_kernel_tools/parallel_utils.py index 1f8d21a7a702..459b785e6504 100644 --- a/tests/kernels/moe/modular_kernel_tools/parallel_utils.py +++ b/tests/kernels/moe/modular_kernel_tools/parallel_utils.py @@ -36,7 +36,6 @@ def _set_vllm_config(vllm_config: VllmConfig, world_size: int, rank: int, import tempfile temp_file = tempfile.mkstemp()[1] - set_current_vllm_config(vllm_config) with set_current_vllm_config(vllm_config): init_distributed_environment( world_size=world_size,