22 lines
444 B
Python
22 lines
444 B
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
from test.registered.kv_canary.test_self_e2e_baseline import _BaselineBase
|
|
|
|
from sglang.test.kv_canary.consts import (
|
|
DSV4_POOL_SERVER_ARGS,
|
|
DSV4_POOL_SERVER_ENV,
|
|
)
|
|
|
|
|
|
class TestBaselineDsv4(_BaselineBase):
|
|
__test__ = True
|
|
|
|
model_mode = "dsv4"
|
|
extra_server_args = DSV4_POOL_SERVER_ARGS
|
|
extra_env = DSV4_POOL_SERVER_ENV
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|