fix: replace vectors_count with count api (#588)

qdrant · Apr 16, 2024 · c6322f0 · c6322f0
1 parent cbefb5a
commit c6322f0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qdrant_client/migrate/migrate.py b/qdrant_client/migrate/migrate.py
@@ -139,8 +139,8 @@ def _migrate_collection(
             collection_name, offset=next_offset, limit=batch_size, with_vectors=True
         )
         dest_client.upload_points(collection_name, records, wait=True)  # type: ignore
-    source_client_vectors_count = source_client.get_collection(collection_name).vectors_count
-    dest_client_vectors_count = dest_client.get_collection(collection_name).vectors_count
+    source_client_vectors_count = source_client.count(collection_name).count
+    dest_client_vectors_count = dest_client.count(collection_name).count
     assert (
         source_client_vectors_count == dest_client_vectors_count
     ), f"Migration failed, vectors count are not equal: source vector count {source_client_vectors_count}, dest vector count {dest_client_vectors_count}"