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

Signed-off-by: Zuyi Zhao <zhaozuy@amazon.com>
This commit is contained in:
Zuyi Zhao 2025-12-01 19:56:39 -08:00 committed by GitHub
parent f441d36cee
commit 53bf71b0f0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -45,7 +45,10 @@ def basic_server_with_lora(smollm2_lora_files):
"64", "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: with RemoteOpenAIServer(MODEL_NAME_SMOLLM, args, env_dict=envs) as remote_server:
yield remote_server yield remote_server