Skip to content

Commit

Permalink
Merge pull request #70 from langchain-ai/mattf/update-embedding-support
Browse files Browse the repository at this point in the history
skip nvolveqa_40k compat test when run against local nim
  • Loading branch information
mattf authored Jul 12, 2024
2 parents 7b588ec + 2084ff0 commit 88c51cc
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions libs/ai-endpoints/tests/integration_tests/test_embeddings.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 88c51cc

Please sign in to comment.