Skip to content

Commit 138b26d

Browse files
zhengd-nvdominicshanshan
authored andcommitted
[https://nvbugs/5607238][test] fix working dir in disagg worker test (NVIDIA#8648)
Signed-off-by: zhengd-nv <[email protected]>
1 parent dbc8bef commit 138b26d

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

tests/integration/defs/disaggregated/test_workers.py

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,6 @@
1010
import pytest
1111
import yaml
1212
from defs.conftest import skip_no_hopper
13-
from defs.disaggregated.test_disaggregated_single_gpu import \
14-
model_path as get_model_path
1513
from defs.trt_test_alternative import popen
1614
from transformers import AutoTokenizer
1715

@@ -206,13 +204,10 @@ def __init__(self,
206204
gen_servers: List[str],
207205
req_timeout_secs: int = DEFAULT_TIMEOUT_REQUEST,
208206
server_start_timeout_secs: int = DEFAULT_TIMEOUT_SERVER_START,
209-
model_name: str = "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
210-
model_path: Optional[str] = None):
207+
model_name: str = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"):
211208
super().__init__(ctx_servers, gen_servers, req_timeout_secs,
212209
server_start_timeout_secs)
213-
if model_path is None:
214-
model_path = get_model_path(model_name)
215-
self.tokenizer = AutoTokenizer.from_pretrained(model_path)
210+
self.tokenizer = AutoTokenizer.from_pretrained(model_name)
216211
self.model_name = model_name
217212
self.kv_cache_block_maps: dict[str, KvCacheAwareServerState] = {}
218213
self.kv_cache_event_maps: dict[str, list[dict]] = {}
@@ -489,6 +484,7 @@ def load_default_prompts(disaggregated_example_root: str):
489484
@contextlib.contextmanager
490485
def background_workers(llm_venv, config_file: str, num_ranks: int = None):
491486
cwd = llm_venv.get_working_directory()
487+
os.chdir(cwd)
492488
with open(os.path.join(cwd, 'output_workers.log'), 'w+') as log_file:
493489
workers_proc, ctx_servers, gen_servers = run_disaggregated_workers(
494490
config_file=config_file,
@@ -603,7 +599,6 @@ def test_workers_kv_cache_aware_router_deepseek_v3_lite_bf16(
603599

604600
with background_workers(llm_venv, config_file,
605601
4) as (ctx_servers, gen_servers):
606-
os.chdir(llm_venv.get_working_directory())
607602
tester = KvCacheAwareRouterTester(ctx_servers,
608603
gen_servers,
609604
model_name="DeepSeek-V3-Lite/bf16",

0 commit comments

Comments
 (0)