Skip to content

Commit 3dfbad6

Browse files
committed
Linter fix
1 parent 273a57b commit 3dfbad6

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

tests/unit_tests/v2/test_async_pg_vectorstore.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import os
2-
from unittest.mock import MagicMock, patch
32
import uuid
43
from typing import AsyncIterator, Sequence
4+
from unittest.mock import MagicMock, patch
55

66
import pytest
77
import pytest_asyncio
@@ -217,12 +217,16 @@ async def test_adelete(self, engine: PGEngine, vs: AsyncPGVectorStore) -> None:
217217
assert result == False
218218
await aexecute(engine, f'TRUNCATE TABLE "{DEFAULT_TABLE}"')
219219

220-
@patch('langchain_postgres.v2.async_vectorstore.storage.Client')
220+
@patch("langchain_postgres.v2.async_vectorstore.storage.Client")
221221
async def test_aadd_images(
222-
self, MockStorageClient: MagicMock, engine: PGEngine, image_vs: AsyncPGVectorStore, image_uris: list[str]
222+
self,
223+
MockStorageClient: MagicMock,
224+
engine: PGEngine,
225+
image_vs: AsyncPGVectorStore,
226+
image_uris: list[str],
223227
) -> None:
224228
mock_blob_instance = MagicMock()
225-
fake_image_bytes = b"fake_gcs_image_data" # Differentiated fake data
229+
fake_image_bytes = b"fake_gcs_image_data" # Differentiated fake data
226230
mock_blob_instance.download_as_bytes.return_value = fake_image_bytes
227231

228232
mock_bucket_instance = MagicMock()

0 commit comments

Comments
 (0)