Skip to content

Commit

Permalink
with nvolveqa_40k removed, deprecation test is not relevant
Browse files Browse the repository at this point in the history
  • Loading branch information
mattf committed May 24, 2024
1 parent 4c876f0 commit f6b0662
Showing 1 changed file with 0 additions and 10 deletions.
10 changes: 0 additions & 10 deletions libs/ai-endpoints/tests/integration_tests/test_embeddings.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@

import pytest
import requests_mock
from langchain_core._api import LangChainDeprecationWarning

from langchain_nvidia_ai_endpoints import NVIDIAEmbeddings

Expand All @@ -18,15 +17,6 @@ def test_embed_query(embedding_model: str, mode: dict) -> None:
assert len(output) > 3


def test_embed_query_deprecated(embedding_model: str, mode: dict) -> None:
"""Test NVIDIA embeddings for a single query."""
query = "foo bar"
with pytest.warns(LangChainDeprecationWarning):
embedding = NVIDIAEmbeddings(model=embedding_model).mode(**mode)
output = embedding.embed_query(query)
assert len(output) > 3


async def test_embed_query_async(embedding_model: str, mode: dict) -> None:
"""Test NVIDIA async embeddings for a single query."""
query = "foo bar"
Expand Down

0 comments on commit f6b0662

Please sign in to comment.