Introduce server_fixtures in sglang.test (#14899)
This commit is contained in:
@@ -2,7 +2,9 @@ import unittest
|
||||
from types import SimpleNamespace
|
||||
|
||||
from sglang.test.few_shot_gsm8k import run_eval as run_eval_few_shot_gsm8k
|
||||
from sglang.test.test_disaggregation_utils import TestDisaggregationBase
|
||||
from sglang.test.server_fixtures.disaggregation_fixture import (
|
||||
PDDisaggregationServerBase,
|
||||
)
|
||||
from sglang.test.test_utils import (
|
||||
DEFAULT_MODEL_NAME_FOR_TEST,
|
||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
||||
@@ -10,7 +12,7 @@ from sglang.test.test_utils import (
|
||||
)
|
||||
|
||||
|
||||
class TestDisaggregationPiecewiseCudaGraph(TestDisaggregationBase):
|
||||
class TestDisaggregationPiecewiseCudaGraph(PDDisaggregationServerBase):
|
||||
"""Test piecewise CUDA graph support in disaggregation prefill server"""
|
||||
|
||||
@classmethod
|
||||
|
||||
Reference in New Issue
Block a user