diff --git a/test/manual/hicache/test_disaggregation_hicache.py b/test/manual/hicache/test_disaggregation_hicache.py index 797393f7c817..466a268596a5 100644 --- a/test/manual/hicache/test_disaggregation_hicache.py +++ b/test/manual/hicache/test_disaggregation_hicache.py @@ -23,17 +23,9 @@ class DisaggregationHiCacheBase(TestDisaggregationBase): @classmethod def setUpClass(cls): + super(DisaggregationHiCacheBase, cls).setUpClass() + cls.model = DEFAULT_MODEL_NAME_FOR_TEST - parsed_url = urlparse(DEFAULT_URL_FOR_TEST) - cls.base_host = parsed_url.hostname - base_port = str(parsed_url.port) - cls.lb_port = base_port - cls.prefill_port = f"{int(base_port) + 100}" - cls.decode_port = f"{int(base_port) + 200}" - cls.prefill_url = f"http://{cls.base_host}:{cls.prefill_port}" - cls.decode_url = f"http://{cls.base_host}:{cls.decode_port}" - cls.lb_url = f"http://{cls.base_host}:{cls.lb_port}" - print(f"{cls.base_host=} {cls.lb_port=} {cls.prefill_port=} {cls.decode_port=}") cls.tokenizer = get_tokenizer(cls.model) cls.temp_dir = tempfile.mkdtemp()