purge unnecessary env variable set in deterministic test (#13481)
This commit is contained in:
@@ -8,10 +8,7 @@ import unittest
|
|||||||
|
|
||||||
from sglang.test.test_deterministic_utils import (
|
from sglang.test.test_deterministic_utils import (
|
||||||
COMMON_SERVER_ARGS,
|
COMMON_SERVER_ARGS,
|
||||||
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
|
||||||
DEFAULT_URL_FOR_TEST,
|
|
||||||
TestDeterministicBase,
|
TestDeterministicBase,
|
||||||
popen_launch_server,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
QWEN3_NEXT = "Qwen/Qwen3-Next-80B-A3B-Instruct"
|
QWEN3_NEXT = "Qwen/Qwen3-Next-80B-A3B-Instruct"
|
||||||
@@ -29,26 +26,6 @@ class TestFlashInferDeterministic(TestDeterministicBase):
|
|||||||
args.extend(["--attention-backend", "flashinfer", "--tp", "4"])
|
args.extend(["--attention-backend", "flashinfer", "--tp", "4"])
|
||||||
return args
|
return args
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def setUpClass(cls):
|
|
||||||
import os
|
|
||||||
|
|
||||||
cls.model = cls.get_model()
|
|
||||||
cls.base_url = DEFAULT_URL_FOR_TEST
|
|
||||||
if "--attention-backend" not in cls.get_server_args():
|
|
||||||
raise unittest.SkipTest("Skip the base test class")
|
|
||||||
|
|
||||||
custom_env = os.environ.copy()
|
|
||||||
custom_env["SGLANG_BATCH_INVARIANT_OPS_ENABLE_MM_DEEPGEMM"] = "false"
|
|
||||||
|
|
||||||
cls.process = popen_launch_server(
|
|
||||||
cls.model,
|
|
||||||
cls.base_url,
|
|
||||||
timeout=DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
|
||||||
other_args=cls.get_server_args(),
|
|
||||||
env=custom_env,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class TestTritonDeterministic(TestDeterministicBase):
|
class TestTritonDeterministic(TestDeterministicBase):
|
||||||
@classmethod
|
@classmethod
|
||||||
@@ -62,26 +39,6 @@ class TestTritonDeterministic(TestDeterministicBase):
|
|||||||
args.extend(["--attention-backend", "triton", "--tp", "4"])
|
args.extend(["--attention-backend", "triton", "--tp", "4"])
|
||||||
return args
|
return args
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def setUpClass(cls):
|
|
||||||
import os
|
|
||||||
|
|
||||||
cls.model = cls.get_model()
|
|
||||||
cls.base_url = DEFAULT_URL_FOR_TEST
|
|
||||||
if "--attention-backend" not in cls.get_server_args():
|
|
||||||
raise unittest.SkipTest("Skip the base test class")
|
|
||||||
|
|
||||||
custom_env = os.environ.copy()
|
|
||||||
custom_env["SGLANG_BATCH_INVARIANT_OPS_ENABLE_MM_DEEPGEMM"] = "false"
|
|
||||||
|
|
||||||
cls.process = popen_launch_server(
|
|
||||||
cls.model,
|
|
||||||
cls.base_url,
|
|
||||||
timeout=DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
|
|
||||||
other_args=cls.get_server_args(),
|
|
||||||
env=custom_env,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
|||||||
Reference in New Issue
Block a user