fix: Fix ci

qdrant · Mar 11, 2025 · 8970512 · 8970512
1 parent 7cfedd1
commit 8970512
Showing 1 changed file with 6 additions and 4 deletions.
diff --git 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