skip nvolveqa_40k compat test when run against local nim

langchain-ai · Jul 11, 2024 · 2084ff0 · 2084ff0
1 parent 7b588ec
commit 2084ff0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/libs/ai-endpoints/tests/integration_tests/test_embeddings.py b/libs/ai-endpoints/tests/integration_tests/test_embeddings.py
@@ -102,6 +102,8 @@ def test_embed_documents_truncate(
 
 @pytest.mark.parametrize("nvolveqa_40k", ["playground_nvolveqa_40k", "nvolveqa_40k"])
 def test_embed_nvolveqa_40k_compat(nvolveqa_40k: str, mode: dict) -> None:
+    if mode:
+        pytest.skip("Test only relevant for API Catalog")
     with pytest.warns(UserWarning):
         embedding = NVIDIAEmbeddings(model=nvolveqa_40k, truncate="NONE", **mode)
     text = "nvidia " * 2048