Skip to content

Commit 58e6ca8

Browse files
committed
linter fix
1 parent b8ca760 commit 58e6ca8

File tree

2 files changed

+3
-7
lines changed

2 files changed

+3
-7
lines changed

tests/test_async_vectorstore_search.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,7 @@ async def test_asimilarity_search_scann(self, vs_custom_scann_query_option):
195195
assert len(results) == 1
196196
assert results == [Document(page_content="foo", id=ids[0])]
197197
results = await vs_custom_scann_query_option.asimilarity_search(
198-
"foo", k=1, filter={'mycontent': 'bar'}
198+
"foo", k=1, filter={"mycontent": "bar"}
199199
)
200200
assert results == [Document(page_content="bar", id=ids[1])]
201201

tests/test_vectorstore.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -221,9 +221,7 @@ async def test_aadd_images_store_uri_only(self, engine_sync, image_uris):
221221
results = await afetch(engine_sync, f'SELECT * FROM "{table_name}"')
222222
assert len(results) == len(image_uris)
223223
for i, result_row in enumerate(results):
224-
assert (
225-
result_row[vs._PGVectorStore__vs.content_column] == image_uris[i]
226-
)
224+
assert result_row[vs._PGVectorStore__vs.content_column] == image_uris[i]
227225
uri_embedding = embeddings_service.embed_query(image_uris[i])
228226
image_embedding = image_embedding_service.embed_image([image_uris[i]])[0]
229227
actual_embedding = json.loads(
@@ -278,9 +276,7 @@ async def test_add_images_store_uri_only(self, engine_sync, image_uris):
278276
results = await afetch(engine_sync, (f'SELECT * FROM "{table_name}"'))
279277
assert len(results) == len(image_uris)
280278
for i, result_row in enumerate(results):
281-
assert (
282-
result_row[vs._PGVectorStore__vs.content_column] == image_uris[i]
283-
)
279+
assert result_row[vs._PGVectorStore__vs.content_column] == image_uris[i]
284280
uri_embedding = embeddings_service.embed_query(image_uris[i])
285281
image_embedding = image_embedding_service.embed_image([image_uris[i]])[0]
286282
actual_embedding = json.loads(

0 commit comments

Comments
 (0)