From 83cbcbc99238bb755dd629df88b78b9f7cb21c3b Mon Sep 17 00:00:00 2001
From: ThreadDao <yufen.zong@zilliz.com>
Date: Fri, 1 Oct 2021 09:00:08 +0800
Subject: [PATCH] Optmize collection wrapper import (#8988)

Signed-off-by: ThreadDao <yufen.zong@zilliz.com>
---
 tests/python_client/base/collection_wrapper.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/python_client/base/collection_wrapper.py b/tests/python_client/base/collection_wrapper.py
index f7b99c8464..5625af2c83 100644
--- a/tests/python_client/base/collection_wrapper.py
+++ b/tests/python_client/base/collection_wrapper.py
@@ -1,8 +1,9 @@
 import sys
 
+from pymilvus import Collection
+
 sys.path.append("..")
-from check.param_check import *
-from check.func_check import *
+from check.func_check import ResponseChecker
 from utils.api_request import api_request
 from utils.util_log import test_log as log