Skip to content

Commit 42c0dc0

Browse files
committed
linter fix
1 parent e30b3df commit 42c0dc0

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
@@ -193,7 +193,7 @@ async def test_asimilarity_search_scann(self, vs_custom_scann_query_option):
193193
assert len(results) == 1
194194
assert results == [Document(page_content="foo", id=ids[0])]
195195
results = await vs_custom_scann_query_option.asimilarity_search(
196-
"foo", k=1, filter={'mycontent': 'bar'}
196+
"foo", k=1, filter={"mycontent": "bar"}
197197
)
198198
assert results == [Document(page_content="bar", id=ids[1])]
199199

tests/test_vectorstore.py

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

0 commit comments

Comments
 (0)