diff --git a/tests/integration/defs/disaggregated/test_auto_scaling.py b/tests/integration/defs/disaggregated/test_auto_scaling.py index 038049109d..4821ee8476 100644 --- a/tests/integration/defs/disaggregated/test_auto_scaling.py +++ b/tests/integration/defs/disaggregated/test_auto_scaling.py @@ -2,7 +2,6 @@ import asyncio import os import shutil import subprocess -import sys import tempfile import traceback import uuid @@ -187,14 +186,13 @@ def run_disagg_server(disagg_cluster_config, work_dir, port=0, save_log=False): cmds = ["trtllm-serve", "disaggregated", "-c", disagg_server_config_path] log_file = None log_path = None + stdout = None + stderr = None if save_log: log_path = os.path.join(work_dir, "disagg_server.log") log_file = open(log_path, "w+") stdout = log_file stderr = log_file - else: - stdout = sys.stdout - stderr = sys.stderr p = subprocess.Popen(cmds, stdout=stdout, stderr=stderr) return ProcessWrapper(p, log_file=log_file, log_path=log_path, port=port)