Infer ARRAY datatype from list like non numeric objects

Signed-off-by: Abhay Ratnaparkhi <[email protected]>
milvus-io · Nov 15, 2023 · 69cbc33 · 69cbc33
1 parent 1428201
commit 69cbc33
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pymilvus/orm/types.py b/pymilvus/orm/types.py
@@ -121,7 +121,10 @@ 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
+            if d_type != DataType.UNKNOWN:
+                return DataType.FLOAT_VECTOR if is_numeric_datatype(d_type) else DataType.ARRAY
+            else:
+                return d_type
 
     if d_type == DataType.UNKNOWN:
         try: