Skip to content

Commit

Permalink
fix: fix typo in splade model name (#754)
Browse files Browse the repository at this point in the history
  • Loading branch information
joein committed Aug 26, 2024
1 parent d3643ca commit e782394
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_fastembed.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ def test_hybrid_query():
if not local_client._FASTEMBED_INSTALLED:
pytest.skip("FastEmbed is not installed, skipping test")

local_client.set_sparse_model(embedding_model_name="prithvida/Splade_PP_en_v1")
local_client.set_sparse_model(embedding_model_name="prithivida/Splade_PP_en_v1")

local_client.add(collection_name=collection_name, **DOCS_EXAMPLE)

Expand Down Expand Up @@ -94,7 +94,7 @@ def test_query_batch():
assert len(dense_search_result) == len(query_texts)
assert all(len(result) > 0 for result in dense_search_result)

local_client.set_sparse_model(embedding_model_name="prithvida/Splade_PP_en_v1")
local_client.set_sparse_model(embedding_model_name="prithivida/Splade_PP_en_v1")

local_client.add(collection_name=hybrid_collection_name, **DOCS_EXAMPLE)

Expand Down

0 comments on commit e782394

Please sign in to comment.