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

[sip e2e] Add new func get_query_segment_info of client file (#14151)

Browse Source 
Signed-off-by: wangting0128 <ting.wang@zilliz.com>
This commit is contained in:
wt 2021-12-24 15:44:30 +08:00 committed by GitHub
parent c32e964978
commit 989c61dd72
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/benchmark/milvus_benchmark/client.py
View File

@ -490,6 +490,12 @@ class MilvusClient(object):
collection_name = self._collection_name
return self._milvus.release_partitions(collection_name, tag_names, timeout=timeout)
@time_wrapper
def get_query_segment_info(self, collection_name=None, timeout=300, **kwargs):
if collection_name is None:
collection_name = self._collection_name
return self._milvus.get_query_segment_info(collection_name, timeout=timeout, **kwargs)
@time_wrapper
def scene_test(self, collection_name=None, vectors=None, ids=None):
"""