purge unnecessary env variable set in deterministic test (#13481)

This commit is contained in:
Minglei Zhu
2025-11-18 20:43:51 -08:00
committed by GitHub
parent 075ba74dd4
commit 3798055929

View File

@@ -8,10 +8,7 @@ import unittest
from sglang.test.test_deterministic_utils import (
COMMON_SERVER_ARGS,
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
DEFAULT_URL_FOR_TEST,
TestDeterministicBase,
popen_launch_server,
)
QWEN3_NEXT = "Qwen/Qwen3-Next-80B-A3B-Instruct"
@@ -29,26 +26,6 @@ class TestFlashInferDeterministic(TestDeterministicBase):
args.extend(["--attention-backend", "flashinfer", "--tp", "4"])
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):
@classmethod
@@ -62,26 +39,6 @@ class TestTritonDeterministic(TestDeterministicBase):
args.extend(["--attention-backend", "triton", "--tp", "4"])
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__":
unittest.main()