From 0b4608f71eb92177b23b5168b9cc9242f257eb6a Mon Sep 17 00:00:00 2001 From: Bagatur <22008038+baskaryan@users.noreply.github.com> Date: Wed, 7 Aug 2024 10:51:39 -0700 Subject: [PATCH] infra: temp skip oai embeddings test (#25148) --- .../openai/tests/integration_tests/embeddings/test_base.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/libs/partners/openai/tests/integration_tests/embeddings/test_base.py b/libs/partners/openai/tests/integration_tests/embeddings/test_base.py index 899a1cd4289..b7a4aa7c8c7 100644 --- a/libs/partners/openai/tests/integration_tests/embeddings/test_base.py +++ b/libs/partners/openai/tests/integration_tests/embeddings/test_base.py @@ -2,6 +2,7 @@ import numpy as np import openai +import pytest from langchain_openai.embeddings.base import OpenAIEmbeddings @@ -32,6 +33,7 @@ def test_langchain_openai_embeddings_dimensions() -> None: assert len(output[0]) == 128 +@pytest.mark.skip(reason="flaky") def test_langchain_openai_embeddings_equivalent_to_raw() -> None: documents = ["disallowed special token '<|endoftext|>'"] embedding = OpenAIEmbeddings() @@ -46,6 +48,7 @@ def test_langchain_openai_embeddings_equivalent_to_raw() -> None: assert np.isclose(lc_output, direct_output).all() +@pytest.mark.skip(reason="flaky") async def test_langchain_openai_embeddings_equivalent_to_raw_async() -> None: documents = ["disallowed special token '<|endoftext|>'"] embedding = OpenAIEmbeddings()