Skip to content

Commit 61076d9

Browse files
committed
Fix trt tests
1 parent 87ca9bc commit 61076d9

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

python/openai/tests/test_embeddings.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -421,9 +421,11 @@
421421
@pytest.mark.fastapi
422422
class TestEmbeddings:
423423
@pytest.fixture(scope="class", autouse=True)
424-
def check_backend(self, backend: str):
424+
def check_backend(backend: str):
425425
if backend != "vllm":
426-
pytest.skip("These tests only run with vLLM backend")
426+
pytest.skip(
427+
"TRT-LLM backend and Python backend do not support embedding requests"
428+
)
427429

428430
@pytest.fixture(scope="class")
429431
def client(self, fastapi_client_class_scope):

0 commit comments

Comments
 (0)