forked from Archives/langchain
Make Integration Tests "work" again (#106)
This fixes Issue #104 The tests for HF Embeddings is skipped because of the segfault issue mentioned there. Perhaps, a new issue should be created for that?
This commit is contained in:
parent
abe4fc04fa
commit
95dd2f140e
@ -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"
|
||||
|
@ -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):
|
||||
|
Loading…
Reference in New Issue
Block a user