From 897051249fe683ef07612b3b97598fad0a13fb1d Mon Sep 17 00:00:00 2001 From: hh-space-invader Date: Tue, 11 Mar 2025 17:09:40 +0200 Subject: [PATCH] fix: Fix ci --- tests/test_async_qdrant_client.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/tests/test_async_qdrant_client.py b/tests/test_async_qdrant_client.py index 9125c3d7..81146e07 100644 --- a/tests/test_async_qdrant_client.py +++ b/tests/test_async_qdrant_client.py @@ -64,6 +64,7 @@ async def test_async_qdrant_client(prefer_grpc): ) assert (await client.get_aliases()).aliases == [] + k = 50 await client.upsert( collection_name=COLLECTION_NAME, points=[ @@ -72,10 +73,10 @@ async def test_async_qdrant_client(prefer_grpc): vector=np.random.rand(10).tolist(), payload={"random_dig": random.randint(1, 100)}, ) - for i in range(50) + for i in range(k) ], ) - assert (await client.count(COLLECTION_NAME)).count == 100 + assert (await client.count(COLLECTION_NAME)).count == k assert len((await client.scroll(COLLECTION_NAME, limit=2))[0]) == 2 @@ -306,6 +307,7 @@ async def test_async_qdrant_client_local(): ) assert await client.get_aliases() + k = 50 await client.upsert( collection_name=COLLECTION_NAME, points=[ @@ -314,10 +316,10 @@ async def test_async_qdrant_client_local(): vector=np.random.rand(10).tolist(), payload={"random_dig": random.randint(1, 100)}, ) - for i in range(50) + for i in range(k) ], ) - assert (await client.count(COLLECTION_NAME)).count == 100 + assert (await client.count(COLLECTION_NAME)).count == k assert len((await client.scroll(COLLECTION_NAME, limit=2))[0]) == 2