diff --git a/tests/integration_tests/embeddings/test_huggingface.py b/tests/integration_tests/embeddings/test_huggingface.py index 5c031ee0453..e71fbb00667 100644 --- a/tests/integration_tests/embeddings/test_huggingface.py +++ b/tests/integration_tests/embeddings/test_huggingface.py @@ -1,7 +1,10 @@ """Test huggingface embeddings.""" +import unittest + from langchain.embeddings.huggingface import HuggingFaceEmbeddings +@unittest.skip("This test causes a segfault.") def test_huggingface_embedding_documents() -> None: """Test huggingface embeddings.""" documents = ["foo bar"] @@ -11,6 +14,7 @@ def test_huggingface_embedding_documents() -> None: assert len(output[0]) == 768 +@unittest.skip("This test causes a segfault.") def test_huggingface_embedding_query() -> None: """Test huggingface embeddings.""" document = "foo bar" diff --git a/tests/integration_tests/test_faiss.py b/tests/integration_tests/test_faiss.py index 0a42b0d75c3..8df51989cbf 100644 --- a/tests/integration_tests/test_faiss.py +++ b/tests/integration_tests/test_faiss.py @@ -6,7 +6,7 @@ import pytest from langchain.docstore.document import Document from langchain.docstore.in_memory import InMemoryDocstore from langchain.embeddings.base import Embeddings -from langchain.faiss import FAISS +from langchain.vectorstores.faiss import FAISS class FakeEmbeddings(Embeddings):