fix: select a single column consisting of a list of column names(milv…

…us-io#2210) Signed-off-by: lixinguo <[email protected]>
smellthemoon · Aug 2, 2024 · 83850f1 · 83850f1
1 parent dc13969
commit 83850f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymilvus/orm/schema.py b/pymilvus/orm/schema.py
@@ -516,7 +516,7 @@ def check_insert_schema(schema: CollectionSchema, data: Union[List[List], pd.Dat
             raise DataNotMatchException(message=msg)
         columns = list(data.columns)
         columns.remove(schema.primary_field)
-        data = data[[columns]]
+        data = data[columns]
 
     _check_data_schema_cnt(schema, data)
     _check_insert_data(data)