From efd5ac626c9deef60862a4a5c8a3a948005af854 Mon Sep 17 00:00:00 2001 From: wt Date: Thu, 21 Oct 2021 19:28:03 +0800 Subject: [PATCH] [skip ci] Add raise of get_vectors_from_binary func on benchmark (#10345) Signed-off-by: wangting0128 --- tests/benchmark/milvus_benchmark/runners/utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/benchmark/milvus_benchmark/runners/utils.py b/tests/benchmark/milvus_benchmark/runners/utils.py index 2b794d0a68..e8b5f9c4fb 100644 --- a/tests/benchmark/milvus_benchmark/runners/utils.py +++ b/tests/benchmark/milvus_benchmark/runners/utils.py @@ -97,6 +97,8 @@ def get_vectors_from_binary(nq, dimension, data_type): file_name = DEEP_SRC_DATA_DIR + 'query.npy' elif data_type == "binary": file_name = BINARY_SRC_DATA_DIR + 'query.npy' + else: + raise Exception("There is no corresponding file for this data type %s." % str(data_type)) data = np.load(file_name) vectors = data[0:nq].tolist() return vectors