Skip to content

Commit

Permalink
test fix
Browse files Browse the repository at this point in the history
  • Loading branch information
choccccy committed Nov 28, 2023
1 parent 6b59a75 commit 68d71b5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test/embedding/test_pgvector.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ async def test_PGv_embed_many_pacer():


@pytest.mark.asyncio
async def test_PGv_embed_pacer():
async def test_PGv_search_specific():
dummy_model = SentenceTransformer('mock_transformer')
dummy_model.encode.return_value = np.array([1, 2, 3])
print(f'EMODEL: {dummy_model}')
Expand Down Expand Up @@ -172,7 +172,7 @@ async def test_PGv_embed_pacer():

# search table with perfect match
search_string = '[beep] A single lap should be completed each time you hear this sound.'
sim_search = await vDB.search(query_string=search_string, page_numbers=[3], conjunctive=False)
sim_search = await vDB.search(query_string=search_string, query_page_numbers=[3], query_tags=['pacer'], conjunctive=False)
assert sim_search is not None, Exception("No results returned from perfect search")

await vDB.drop_table()

0 comments on commit 68d71b5

Please sign in to comment.