fix

AlibabaPAI · Aug 30, 2024 · 41fe8b2 · 41fe8b2
1 parent 59f3a63
commit 41fe8b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/backends/vllm/test_migration.py b/tests/backends/vllm/test_migration.py
@@ -50,8 +50,8 @@ def __init__(self):
 # @pytest.mark.skipif(torch.cuda.device_count() < 2,
 #                     reason="Need at least 2 GPUs to run the test.")
 @pytest.mark.skip(reason="Regression Test")
+@pytest.fixture
 def test_migration_correctness():
-    ray.init(namespace="llumnix", ignore_reinit_error=True)
     engine_args = EngineArgs(model="facebook/opt-125m",worker_use_ray=True)
     id_rank_map = {"0":0,"1":1}
     migration_config = MigrationConfig("LCFS", "gloo",16,1,4,5,20)