Skip ai21 tests for python3.9

DataDog · Jul 29, 2024 · e7c7e0a · e7c7e0a
1 parent 2c5b38c
commit e7c7e0a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/tests/contrib/langchain/test_langchain_community.py b/tests/contrib/langchain/test_langchain_community.py
@@ -133,7 +133,10 @@ def test_cohere_llm_sync(langchain_cohere, request_vcr):
         llm.invoke("What is the secret Krabby Patty recipe?")
 
 
-@pytest.mark.skipif(LANGCHAIN_VERSION < (0, 2), reason="Requires separate cassette for langchain v0.1")
+@pytest.mark.skipif(
+    LANGCHAIN_VERSION < (0, 2) or sys.version_info < (3, 10),
+    reason="Requires separate cassette for langchain v0.1, Python 3.9",
+)
 @pytest.mark.snapshot
 def test_ai21_llm_sync(langchain_community, request_vcr):
     if langchain_community is None:

diff --git a/tests/contrib/langchain/test_langchain_llmobs.py b/tests/contrib/langchain/test_langchain_llmobs.py
@@ -501,7 +501,7 @@ def test_llmobs_anthropic_chat_model(self, langchain_anthropic, mock_llmobs_span
 
 
 @pytest.mark.skipif(LANGCHAIN_VERSION < (0, 1), reason="These tests are for langchain >= 0.1.0")
-class TestLangchainTraceStructureWithLlmIntegrations(SubprocessTestCase):
+class TestTraceStructureWithLLMIntegrations(SubprocessTestCase):
     bedrock_env_config = dict(
         AWS_ACCESS_KEY_ID="testing",
         AWS_SECRET_ACCESS_KEY="testing",