diff --git a/python/sglang/test/server_fixtures/streaming_session_fixture.py b/python/sglang/test/server_fixtures/streaming_session_fixture.py index 8ffd111b0..d26415b97 100644 --- a/python/sglang/test/server_fixtures/streaming_session_fixture.py +++ b/python/sglang/test/server_fixtures/streaming_session_fixture.py @@ -402,7 +402,7 @@ class StreamingSessionServerBase(CustomTestCase): - `env_overrides`: list of `(env_attr_name, value)` tuples; each is pushed onto the `setUpClass` context stack so the env override is live during `popen_launch_server` and torn down on - `tearDownClass`-time. `SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY=2` + `tearDownClass`-time. `SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY=1` is always applied on top of these. """ @@ -417,7 +417,7 @@ class StreamingSessionServerBase(CustomTestCase): with contextlib.ExitStack() as stack: stack.enter_context( - envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY.override(2) + envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY.override(1) ) for name, val in cls.env_overrides: stack.enter_context(getattr(envs, name).override(val)) diff --git a/test/registered/scheduler/test_mixed_chunked_prefill.py b/test/registered/scheduler/test_mixed_chunked_prefill.py index 15c448d92..6af02f554 100644 --- a/test/registered/scheduler/test_mixed_chunked_prefill.py +++ b/test/registered/scheduler/test_mixed_chunked_prefill.py @@ -29,7 +29,7 @@ class TestMixedChunkedPrefill(GSM8KMixin, CustomTestCase): @classmethod def setUpClass(cls): - with envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY.override(2): + with envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY.override(1): cls.process = popen_launch_server( cls.model, cls.base_url,