diff --git a/tests/entrypoints/test_openai_server.py b/tests/entrypoints/test_openai_server.py index b5ddbe9dafc06..f626ce62ff4b8 100644 --- a/tests/entrypoints/test_openai_server.py +++ b/tests/entrypoints/test_openai_server.py @@ -534,13 +534,6 @@ async def test_guided_json_chat(server, client: openai.AsyncOpenAI): assert json1["name"] != json2["name"] assert json1["age"] != json2["age"] - -# *** UPSTREAM SYNC *** -# This test covers an experimental feature in vLLM, guided generation. -# Currently, there is an upstream issue being debugged -# See: https://github.com/vllm-project/vllm/pull/3383 -# Once this is resolved upstream, turn the test back on. -@pytest.mark.skip("Issue upstream, currently being resolved.") async def test_guided_regex_completion(server, client: openai.AsyncOpenAI): completion = await client.completions.create( model=MODEL_NAME, @@ -552,9 +545,7 @@ async def test_guided_regex_completion(server, client: openai.AsyncOpenAI): assert completion.id is not None assert completion.choices is not None and len(completion.choices) == 3 - for i in range(3): - print(completion.choices[i].text) assert completion.choices[i].text is not None assert re.fullmatch(TEST_REGEX, completion.choices[i].text) is not None