Skip to content

Commit

Permalink
return array data type in infer_dtype_bydata
Browse files Browse the repository at this point in the history
Signed-off-by: Rohit Gupta <rohit.gupta2@walmart.com>
  • Loading branch information
Rohit Gupta committed Jun 20, 2024
1 parent b85144d commit 0e8371e
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions pymilvus/orm/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,6 +123,7 @@ def infer_dtype_bydata(data: Any):
d_type = dtype_str_map.get(type_str, DataType.UNKNOWN)
return DataType.FLOAT_VECTOR if is_numeric_datatype(d_type) else DataType.ARRAY


if d_type == DataType.UNKNOWN:
try:
elem = data[0]
Expand Down

0 comments on commit 0e8371e

Please sign in to comment.