admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

add compact and time count

Browse Source 
Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
This commit is contained in:
zhuwenxing 2024-07-24 10:58:09 +08:00
parent 1ac15a215a
commit 3d8ea03f5d
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/python_client/filter_search/prepare_custom_data.py
View File

@ -155,6 +155,9 @@ def prepare_data(host="127.0.0.1", port=19530, user=None, pwd=None, minio_host="
logger.info(f"insert {len(data)} cost time {time.time() - t0}")
else:
raise ValueError(f"insert_mode {insert_mode} not supported")
collection.compact()
collection.wait_for_compaction_completed()
t0 = time.time()
collection.create_index("emb", index_params=index_params)
index_list = utility.list_indexes(collection_name=collection_name)
for index_name in index_list:
@ -164,7 +167,13 @@ def prepare_data(host="127.0.0.1", port=19530, user=None, pwd=None, minio_host="
progress = utility.index_building_progress(collection_name=collection_name, index_name=index_name)
logger.info(f"collection {collection_name} index {index_name} progress: {progress}")
logger.info(f"collection {collection_name} index {index_name} progress: {progress}")
tt = time.time() - t0
logger.info(f"create index cost time {tt}")
collection.compact()
collection.wait_for_compaction_completed()
t0 = time.time()
collection.load()
logger.info(f"load collection cost time {time.time() - t0}")
num = collection.num_entities
logger.info(f"collection {collection_name} loaded, num_entities: {num}")