From 77c9a8daa11e5eada19a54156eae0e987c7fcfee Mon Sep 17 00:00:00 2001 From: wt Date: Thu, 7 Oct 2021 18:03:15 +0800 Subject: [PATCH] [skip ci] Modify the conditions for checking the existence of milvus (#9403) Signed-off-by: wangting0128 --- tests/benchmark/milvus_benchmark/client.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tests/benchmark/milvus_benchmark/client.py b/tests/benchmark/milvus_benchmark/client.py index dadfd5c73e..6f617c4fcf 100644 --- a/tests/benchmark/milvus_benchmark/client.py +++ b/tests/benchmark/milvus_benchmark/client.py @@ -401,8 +401,13 @@ class MilvusClient(object): i = 0 while i < timeout: try: - row_count = self.count(collection_name=collection_name) - if row_count: + # row_count = self.count(collection_name=collection_name) + # if row_count: + # time.sleep(1) + # i = i + 1 + # continue + res = self._milvus.has_collection(collection_name) + if res: time.sleep(1) i = i + 1 continue