From 989c61dd72a017c9836d1e296502b8f12307a98a Mon Sep 17 00:00:00 2001 From: wt Date: Fri, 24 Dec 2021 15:44:30 +0800 Subject: [PATCH] [sip e2e] Add new func get_query_segment_info of client file (#14151) Signed-off-by: wangting0128 --- tests/benchmark/milvus_benchmark/client.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/benchmark/milvus_benchmark/client.py b/tests/benchmark/milvus_benchmark/client.py index fe46c331ef..2271508af8 100644 --- a/tests/benchmark/milvus_benchmark/client.py +++ b/tests/benchmark/milvus_benchmark/client.py @@ -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): """