Skip to content

Commit 5fdaeef

Browse files
authored
Infer ARRAY datatype from list like non numeric objects
1 parent 1428201 commit 5fdaeef

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pymilvus/orm/types.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,10 @@ def infer_dtype_bydata(data: Any):
121121
failed = True
122122
if not failed:
123123
d_type = dtype_str_map.get(type_str, DataType.UNKNOWN)
124-
return DataType.FLOAT_VECTOR if is_numeric_datatype(d_type) else DataType.UNKNOWN
124+
if d_type != DataType.UNKNOWN:
125+
return DataType.FLOAT_VECTOR if is_numeric_datatype(d_type) else DataType.ARRAY
126+
else:
127+
return d_type
125128

126129
if d_type == DataType.UNKNOWN:
127130
try:

0 commit comments

Comments
 (0)