diff --git a/qdrant_client/local/local_collection.py b/qdrant_client/local/local_collection.py index f14766d2..190edf5d 100644 --- a/qdrant_client/local/local_collection.py +++ b/qdrant_client/local/local_collection.py @@ -1599,7 +1599,7 @@ def info(self) -> models.CollectionInfo: return models.CollectionInfo( status=models.CollectionStatus.GREEN, optimizer_status=models.OptimizersStatusOneOf.OK, - vectors_count=self.count().count * len(self.vectors), + vectors_count=None, indexed_vectors_count=0, # LocalCollection does not do indexing points_count=self.count().count, segments_count=1, diff --git a/tests/congruence_tests/test_common.py b/tests/congruence_tests/test_common.py index d6432043..f88cf0db 100644 --- a/tests/congruence_tests/test_common.py +++ b/tests/congruence_tests/test_common.py @@ -122,7 +122,7 @@ def compare_collections( client_1, client_2, num_vectors, - attrs=("vectors_count", "indexed_vectors_count", "points_count"), + attrs=("indexed_vectors_count", "points_count"), collection_name: str = COLLECTION_NAME, ): collection_1 = client_1.get_collection(collection_name) diff --git a/tests/congruence_tests/test_sparse_updates.py b/tests/congruence_tests/test_sparse_updates.py index d93f6e9f..c2b785a5 100644 --- a/tests/congruence_tests/test_sparse_updates.py +++ b/tests/congruence_tests/test_sparse_updates.py @@ -92,10 +92,7 @@ def test_upsert(): local_client, remote_client, UPLOAD_NUM_VECTORS, - attrs=( - "points_count", - "vectors_count", - ), + attrs=("points_count",), ) @@ -122,10 +119,7 @@ def test_upload_collection(): local_client, remote_client, UPLOAD_NUM_VECTORS, - attrs=( - "points_count", - "vectors_count", - ), + attrs=("points_count",), ) @@ -155,10 +149,7 @@ def test_upload_collection_generators(): local_client, remote_client, UPLOAD_NUM_VECTORS, - attrs=( - "points_count", - "vectors_count", - ), + attrs=("points_count",), ) @@ -178,10 +169,7 @@ def test_upload_points(): local_client, remote_client, UPLOAD_NUM_VECTORS, - attrs=( - "points_count", - "vectors_count", - ), + attrs=("points_count",), ) @@ -213,8 +201,5 @@ def test_upload_uuid_in_batches(): local_client, remote_client, UPLOAD_NUM_VECTORS, - attrs=( - "points_count", - "vectors_count", - ), + attrs=("points_count",), ) diff --git a/tests/test_async_qdrant_client.py b/tests/test_async_qdrant_client.py index e45028e7..51a8449f 100644 --- a/tests/test_async_qdrant_client.py +++ b/tests/test_async_qdrant_client.py @@ -97,7 +97,7 @@ async def test_async_grpc(): @pytest.mark.parametrize("prefer_grpc", [True, False]) async def test_async_qdrant_client(prefer_grpc): version = os.getenv("QDRANT_VERSION") - client = AsyncQdrantClient(prefer_grpc=prefer_grpc) + client = AsyncQdrantClient(prefer_grpc=prefer_grpc, timeout=15) collection_params = dict( collection_name=COLLECTION_NAME, vectors_config=models.VectorParams(size=10, distance=models.Distance.EUCLID), @@ -244,10 +244,7 @@ async def test_async_qdrant_client(prefer_grpc): # await client.recover_snapshot(collection_name=COLLECTION_NAME, location=...) # assert (await client.get_collection(COLLECTION_NAME)).vectors_count == 100 - await client.delete_snapshot(COLLECTION_NAME, snapshot_name=snapshots[0].name) - time.sleep( - 0.5 - ) # wait param is not propagated https://github.com/qdrant/qdrant-client/issues/254 + await client.delete_snapshot(COLLECTION_NAME, snapshot_name=snapshots[0].name, wait=True) assert len(await client.list_snapshots(COLLECTION_NAME)) == 0 @@ -255,10 +252,7 @@ async def test_async_qdrant_client(prefer_grpc): snapshots = await client.list_full_snapshots() assert len(snapshots) == 1 - await client.delete_full_snapshot(snapshot_name=snapshots[0].name) - time.sleep( - 0.5 - ) # wait param is not propagated https://github.com/qdrant/qdrant-client/issues/254 + await client.delete_full_snapshot(snapshot_name=snapshots[0].name, wait=True) assert len(await client.list_full_snapshots()) == 0