Skip to content

Commit

Permalink
add array data type for milvus vector store collection create
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 27, 2024
1 parent 76008d9 commit f61280d
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions pymilvus/orm/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,7 @@ def infer_dtype_by_scalar_data(data: Any):
return DataType.UNKNOWN


def infer_dtype_bydata(data: Any, **kargs):
def infer_dtype_bydata(data: Any, **kwargs):
d_type = DataType.UNKNOWN
if is_scalar(data):
return infer_dtype_by_scalar_data(data)
Expand All @@ -131,11 +131,11 @@ def infer_dtype_bydata(data: Any, **kargs):
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 kargs is None or len(kargs) == 0:
if kwargs is None or len(kargs) == 0:
return DataType.FLOAT_VECTOR if \
is_numeric_datatype(d_type) else DataType.UNKNOWN
else:
if kargs["type"] is not None and kargs["type"] == "vector":
if kwargs["type"] is not None and kwargs["type"] == "vector":
return DataType.FLOAT_VECTOR \
if is_numeric_datatype(d_type) else DataType.UNKNOWN
else:
Expand Down

0 comments on commit f61280d

Please sign in to comment.