From 64262f0f12351ca3fbd14e206c68fe4a4f496770 Mon Sep 17 00:00:00 2001 From: Mrinal Chanshetty Date: Wed, 21 Oct 2026 01:07:01 -0700 Subject: [PATCH] test: exempt the providers mirrored from main in the cost-map provider registry test (cherry picked from commit 9c28744ad56325455b54db8a9a9b5a683f73e5f1, tests/unit/test_model_prices_schema.py only) --- tests/unit/test_model_prices_schema.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/tests/unit/test_model_prices_schema.py b/tests/unit/test_model_prices_schema.py index 86a3a1e6a57..8c55a326655 100644 --- a/tests/unit/test_model_prices_schema.py +++ b/tests/unit/test_model_prices_schema.py @@ -511,7 +511,19 @@ def test_deepseek_rows_bill_half_rate_outside_weekday_peak_hours(path: Path): assert all(name in rows for name in DEEPSEEK_PRICED_ROWS) -PROVIDER_LABELS_WITHOUT_A_MODEL_SET: Final = frozenset({"sagemaker", "bedrock_converse"}) +PROVIDER_LABELS_WITHOUT_A_MODEL_SET: Final = frozenset( + { + "sagemaker", + "bedrock_converse", + "coralbricks", + "inferenceapis", + "microsoft_365_copilot", + "openinfer", + "prism", + "sail", + "scaledown", + } +) MODES_SERVED_OUTSIDE_THE_LLM_PROVIDER_REGISTRY: Final = frozenset({"search", "evaluation"}) VERTEX_FAMILIES_A_VERTEX_WILDCARD_GRANT_DOES_NOT_LIST: Final = frozenset( {