diff --git a/tests/integration/inference/test_text_inference.py b/tests/integration/inference/test_text_inference.py index d7da6ff86..a137d67cc 100644 --- a/tests/integration/inference/test_text_inference.py +++ b/tests/integration/inference/test_text_inference.py @@ -24,13 +24,14 @@ def skip_if_model_doesnt_support_completion(client_with_models, model_id): providers = {p.provider_id: p for p in client_with_models.providers.list()} provider = providers[provider_id] if ( - provider.provider_type - in ( - "remote::openai", - "remote::anthropic", - "remote::gemini", - "remote::groq", - "remote::sambanova", + provider.provider_type + in ( + "remote::openai", + "remote::anthropic", + "remote::gemini", + "remote::groq", + "remote::sambanova", + ) or "openai-compat" in provider.provider_type ): pytest.skip(f"Model {model_id} hosted by {provider.provider_type} doesn't support completion")