fix

AlibabaPAI · Sep 12, 2024 · c48d69c · c48d69c
1 parent 28ad93f
commit c48d69c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/e2e_test/test_e2e.py b/tests/e2e_test/test_e2e.py
@@ -112,7 +112,7 @@ async def test_e2e(model):
 
     llumnix_output = {}
     for prompt in prompts:
-        response = await asyncio.wait_for(get_llumnix_responce(prompt, sampling_params, "127.0.0.1:37000"),
+        response = await asyncio.wait_for(get_llumnix_responce(prompt, sampling_params, "127.0.0.1:37037"),
                                           timeout=60*5)
         llumnix_output[prompt] = response['text'][0]
 

diff --git a/tests/e2e_test/test_migration.py b/tests/e2e_test/test_migration.py
@@ -82,8 +82,8 @@ async def run_bench_command(command):
     averger_speed = parse_log_file(instance_output_logs)
 
     data = [
-        ['migration_size'].extend(list(averger_speed.keys())),
-        ['speed(GB/s)'].extend([f"{value:.2f}" for value in averger_speed.values()])
+        ['migration_size'] + list(averger_speed.keys()),
+        ['speed(GB/s)'] + [f"{value:.2f}" for value in averger_speed.values()]
     ]
 
     with open("performance.txt", "w") as f: