[skip ci] Add raise of get_vectors_from_binary func on benchmark (#10345)

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
pull/10384/head
wt 2021-10-21 19:28:03 +08:00 committed by GitHub
parent c1ae837231
commit efd5ac626c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions

View File

@ -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