test: wait all tasks finished in import test (#33674)

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
2024-06-07 10:35:12 +08:00 · 2024-06-07 10:35:12 +08:00 · e3d50a192d
parent 19936d9994
commit e3d50a192d
1 changed files with 1 additions and 1 deletions
--- a/tests/python_client/testcases/test_bulk_insert.py
+++ b/tests/python_client/testcases/test_bulk_insert.py
@ -1493,7 +1493,7 @@ class TestBulkInsert(TestcaseBaseBulkInsert):
            )
            task_ids.append(task_id)
        success, states = self.utility_wrap.wait_for_bulk_insert_tasks_completed(
-            task_ids=[task_id], timeout=300
+            task_ids=task_ids, timeout=300
        )
        log.info(f"bulk insert state:{success}")