mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-02 03:26:17 +00:00
nits (#210)
use json.dump move test to integration tests (since it requires huggingface_hub)
This commit is contained in:
@@ -1,4 +1,6 @@
|
||||
"""Test HuggingFaceHub embeddings."""
|
||||
import pytest
|
||||
|
||||
from langchain.embeddings import HuggingFaceHubEmbeddings
|
||||
|
||||
|
||||
@@ -17,3 +19,10 @@ def test_huggingfacehub_embedding_query() -> None:
|
||||
embedding = HuggingFaceHubEmbeddings()
|
||||
output = embedding.embed_query(document)
|
||||
assert len(output) == 768
|
||||
|
||||
|
||||
def test_huggingfacehub_embedding_invalid_repo() -> None:
|
||||
"""Test huggingfacehub embedding repo id validation."""
|
||||
# Only sentence-transformers models are currently supported.
|
||||
with pytest.raises(ValueError):
|
||||
HuggingFaceHubEmbeddings(repo_id="allenai/specter")
|
||||
|
Reference in New Issue
Block a user