reformatted

milvus-io · Jul 5, 2024 · 12a14d8 · 12a14d8
1 parent 727d14e
commit 12a14d8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pymilvus/orm/types.py b/pymilvus/orm/types.py
@@ -131,9 +131,7 @@ def infer_dtype_bydata(data: Any, **kwargs):
             d_type = dtype_str_map.get(type_str, DataType.UNKNOWN)
             if is_varchar_datatype(d_type) or is_bool_datatype(d_type):
                 return DataType.ARRAY
-            if kwargs is None or len(kwargs) == 0:
-                return DataType.FLOAT_VECTOR if is_numeric_datatype(d_type) else DataType.UNKNOWN
-            elif kwargs["type"] is not None and kwargs["type"] == "vector":
+            if kwargs is None or len(kwargs) == 0 or (kwargs["type"] is not None and kwargs["type"] == "vector"):
                 return (
                     DataType.FLOAT_VECTOR if is_numeric_datatype(d_type) else DataType.UNKNOWN
                 )