Fix test migration

AlibabaPAI · Oct 22, 2024 · ec89f07 · ec89f07
1 parent 7c82c5a
commit ec89f07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/e2e_test/test_migration.py b/tests/e2e_test/test_migration.py
@@ -67,7 +67,7 @@ def parse_manager_log_file(log_file):
 @pytest.mark.parametrize("model", ['/mnt/model/Qwen-7B'])
 @pytest.mark.parametrize("migration_backend", ['rpc', 'gloo', 'nccl'])
 @pytest.mark.parametrize("migrated_request_status", ['running', 'waiting'])
-async def test_migration_benchmark(model, migration_backend):
+async def test_migration_benchmark(model, migration_backend, migrated_request_status):
     if migrated_request_status == 'waiting' and migration_backend != 'rpc':
         pytest.skip("When the migrated request status is waiting, only test the rpc migration backend.")