Remove AzurePhiModel from tests

parkervg · Oct 22, 2024 · 9885886 · 9885886
1 parent 847bcc1
commit 9885886
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -10,7 +10,6 @@
     OllamaLLM,
     OpenaiLLM,
     AnthropicLLM,
-    AzurePhiModel,
     Model,
 )
 from blendsql import LLMQA, LLMMap, LLMJoin
@@ -57,8 +56,8 @@ def pytest_generate_tests(metafunc):
             model_list.append(AnthropicLLM("claude-3-haiku-20240307", caching=False))
 
         # Azure Phi check
-        if all(os.getenv(k) is not None for k in ["AZURE_PHI_KEY", "AZURE_PHI_URL"]):
-            model_list.append(AzurePhiModel(caching=False))
+        # if all(os.getenv(k) is not None for k in ["AZURE_PHI_KEY", "AZURE_PHI_URL"]):
+        #     model_list.append(AzurePhiModel(caching=False))
 
         metafunc.parametrize("model", model_list)