milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Change to unique primary keys for inserting data into partitions (#10209) 

Signed-off-by: Binbin Lv <binbin.lv@zilliz.com>
pull/10221/head
binbin 2021-10-19 20:20:56 +08:00 committed by GitHub
parent 8daeb0d519
commit 7bba7d96c5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/python_client/common/common_func.py
View File

@ -496,10 +496,11 @@ def insert_data(collection_w, nb=3000, is_binary=False, is_all_data_type=False,
vectors = []
binary_raw_vectors = []
insert_ids = []
start = 0
log.info("insert_data: inserting data into collection %s (num_entities: %s)"
% (collection_w.name, nb))
for i in range(num):
default_data = gen_default_dataframe_data(nb // num, dim=dim)
default_data = gen_default_dataframe_data(nb // num, dim=dim, start=start)
if is_binary:
default_data, binary_raw_data = gen_default_binary_dataframe_data(nb // num, dim=dim)
binary_raw_vectors.extend(binary_raw_data)
@ -510,6 +511,7 @@ def insert_data(collection_w, nb=3000, is_binary=False, is_all_data_type=False,
insert_res = collection_w.insert(default_data, par[i].name)[0]
insert_ids.extend(insert_res.primary_keys)
vectors.append(default_data)
start = start + nb // num
log.info("insert_data: inserted data into collection %s (num_entities: %s)"
% (collection_w.name, nb))
return collection_w, vectors, binary_raw_vectors, insert_ids