update

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
2 years ago · f46ffe7b50
1 changed files with 6 additions and 3 deletions
--- a/milvus_client.py
+++ b/milvus_client.py
@ -22,7 +22,7 @@ class MilvusClient(PyOperator):

    def __call__(self, *data):
        """
-        Insert data to Milvus.
+        Insert one row to Milvus.

        Args:
        data (`list`):
@ -34,7 +34,10 @@ class MilvusClient(PyOperator):
        """
        row = []
        for item in data:
-            row.append([item])
+            if isinstance(item, list):
+                row.extend([[i] for i in item])
+            else:
+                row.append([item])
        mr = self._collection.insert(row)
        if mr.err_count > 0:
            raise RuntimeError("Insert to milvus failed")