return array data type in infer_dtype_bydata

milvus-io · Jun 20, 2024 · 0ad77c5 · 0ad77c5
1 parent f7a4839
commit 0ad77c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymilvus/orm/types.py b/pymilvus/orm/types.py
@@ -121,7 +121,7 @@ def infer_dtype_bydata(data: Any):
             failed = True
         if not failed:
             d_type = dtype_str_map.get(type_str, DataType.UNKNOWN)
-            return DataType.FLOAT_VECTOR if is_numeric_datatype(d_type) else DataType.UNKNOWN
+            return DataType.FLOAT_VECTOR if is_numeric_datatype(d_type) else DataType.ARRAY
 
     if d_type == DataType.UNKNOWN:
         try: