From c48d69c45ee3941d876677fe6da5be89dee2fac0 Mon Sep 17 00:00:00 2001 From: root Date: Thu, 12 Sep 2024 09:20:12 +0000 Subject: [PATCH] fix --- tests/e2e_test/test_e2e.py | 2 +- tests/e2e_test/test_migration.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/e2e_test/test_e2e.py b/tests/e2e_test/test_e2e.py index f62196d0..e902fa83 100644 --- 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 index 87c388b3..35464f22 100644 --- 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: