phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-09 19:58:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: Enable logprobs top_k tests for remote::vllm

Browse source 
Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
This commit is contained in:
Yuan Tang 2025-02-13 13:40:56 -05:00
parent 8ff27b58fa
commit 1dd24460c8
No known key found for this signature in database
1 changed files with 1 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/client-sdk/inference/test_text_inference.py
View file

@ -14,13 +14,7 @@ PROVIDER_TOOL_PROMPT_FORMAT = {
"remote::vllm": "json",
}
PROVIDER_LOGPROBS_TOP_K = set(
{
"remote::together",
"remote::fireworks",
# "remote:vllm"
}
)
PROVIDER_LOGPROBS_TOP_K = {"remote::together", "remote::fireworks", "remote::vllm"}
@pytest.fixture(scope="session")