From 1a741d1b88d010933bae9b7e5597fe5959002990 Mon Sep 17 00:00:00 2001 From: XuanYang-cn Date: Tue, 23 Apr 2024 19:25:21 +0800 Subject: [PATCH] fix: Remove np.view for floatvector (#2048) Signed-off-by: yangxuan --- examples/hello_milvus.py | 9 ++++++++- pymilvus/client/entity_helper.py | 2 +- pymilvus/orm/prepare.py | 2 +- 3 files changed, 10 insertions(+), 3 deletions(-) diff --git a/examples/hello_milvus.py b/examples/hello_milvus.py index 0cbb8f944..795db825d 100644 --- a/examples/hello_milvus.py +++ b/examples/hello_milvus.py @@ -73,11 +73,18 @@ # provide the pk field because `auto_id` is set to False [str(i) for i in range(num_entities)], rng.random(num_entities).tolist(), # field random, only supports list - rng.random((num_entities, dim)), # field embeddings, supports numpy.ndarray and list + rng.random((num_entities, dim), np.float32), # field embeddings, supports numpy.ndarray and list ] insert_result = hello_milvus.insert(entities) +row = { + "pk": "19530", + "random": 0.5, + "embeddings": rng.random((1, dim), np.float32)[0] +} +hello_milvus.insert(row) + hello_milvus.flush() print(f"Number of entities in Milvus: {hello_milvus.num_entities}") # check the num_entities diff --git a/pymilvus/client/entity_helper.py b/pymilvus/client/entity_helper.py index 47a224670..8d39acc75 100644 --- a/pymilvus/client/entity_helper.py +++ b/pymilvus/client/entity_helper.py @@ -312,7 +312,7 @@ def pack_field_value_to_field_data( raise ParamError( message="invalid input for float32 vector, expect np.ndarray with dtype=float32" ) - f_value = field_value.view(np.float32).tolist() + f_value = field_value.tolist() field_data.vectors.dim = len(f_value) field_data.vectors.float_vector.data.extend(f_value) diff --git a/pymilvus/orm/prepare.py b/pymilvus/orm/prepare.py index 5e2902a20..799fcad04 100644 --- a/pymilvus/orm/prepare.py +++ b/pymilvus/orm/prepare.py @@ -91,7 +91,7 @@ def prepare_insert_data( field.name, "np.float32/np.float64", f_data.dtype ) ) - d = f_data.view(np.float32).tolist() + d = f_data.tolist() elif isinstance(f_data[0], np.ndarray): for ndarr in f_data: