From 2a19f4927d548193da5e37212d8c4d33bdbe7ce0 Mon Sep 17 00:00:00 2001 From: yanliang567 <82361606+yanliang567@users.noreply.github.com> Date: Fri, 10 Feb 2023 17:12:32 +0800 Subject: [PATCH] Fix gen_vectors typo issue (#22117) Signed-off-by: yanliang567 --- tests/python_client/base/client_base.py | 6 ++++-- tests/python_client/common/common_func.py | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/python_client/base/client_base.py b/tests/python_client/base/client_base.py index bf9699e878..39a8d19ac1 100644 --- a/tests/python_client/base/client_base.py +++ b/tests/python_client/base/client_base.py @@ -14,6 +14,8 @@ from utils.util_log import test_log as log from common import common_func as cf from common import common_type as ct +from pymilvus import ResourceGroupInfo + class Base: """ Initialize class object """ @@ -67,8 +69,8 @@ class Base: """ Clean up the rgs before disconnect """ for rg_name in self.resource_group_list: - rg = self.utility_wrap.describe_resource_group(name=rg_name)[0] - if rg is not None: + rg = self.utility_wrap.describe_resource_group(name=rg_name, check_task=ct.CheckTasks.check_nothing)[0] + if isinstance(rg, ResourceGroupInfo): if rg.num_available_node > 0: self.utility_wrap.transfer_node(source=rg_name, target=ct.default_resource_group_name, diff --git a/tests/python_client/common/common_func.py b/tests/python_client/common/common_func.py index a5646c5817..2f5e8fd0b5 100644 --- a/tests/python_client/common/common_func.py +++ b/tests/python_client/common/common_func.py @@ -327,7 +327,7 @@ def gen_default_list_data(nb=ct.default_nb, dim=ct.default_dim, start=0): int_values = [i for i in range(start, start + nb)] float_values = [np.float32(i) for i in range(start, start + nb)] string_values = [str(i) for i in range(start, start + nb)] - float_vec_values = gen_vectors(start + nb, dim) + float_vec_values = gen_vectors(nb, dim) data = [int_values, float_values, string_values, float_vec_values] return data