[Misc] Update conftest for entrypoints/sagemaker test folder (#29799)

Signed-off-by: Zuyi Zhao <zhaozuy@amazon.com>

[Misc] Update conftest for entrypoints/sagemaker test folder (#29799)
Signed-off-by: Zuyi Zhao <zhaozuy@amazon.com>
53bf71b0 · Zuyi Zhao · GitHub · f441d36c · 53bf71b0
Unverified Commit 53bf71b0 authored Dec 01, 2025 by Zuyi Zhao Committed by GitHub Dec 01, 2025
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

tests/entrypoints/sagemaker/conftest.py tests/entrypoints/sagemaker/conftest.py +4 -1

No files found.
--- a/tests/entrypoints/sagemaker/conftest.py
+++ b/tests/entrypoints/sagemaker/conftest.py
@@ -45,7 +45,10 @@ def basic_server_with_lora(smollm2_lora_files):
        "64",
    ]

-    envs = {"VLLM_ALLOW_RUNTIME_LORA_UPDATING": "True"}
+    envs = {
+        "VLLM_ALLOW_RUNTIME_LORA_UPDATING": "True",
+        "SAGEMAKER_ENABLE_STATEFUL_SESSIONS": "True",
+    }
    with RemoteOpenAIServer(MODEL_NAME_SMOLLM, args, env_dict=envs) as remote_server:
        yield remote_server