diff --git a/tests/client-sdk/inference/test_text_inference.py b/tests/client-sdk/inference/test_text_inference.py index 206629602..c931ca255 100644 --- a/tests/client-sdk/inference/test_text_inference.py +++ b/tests/client-sdk/inference/test_text_inference.py @@ -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")