From e2af77afa710d535f4603867aa9d6f3c3f9d75e7 Mon Sep 17 00:00:00 2001
From: Xiangyu Wang <xiangyu.wang@zilliz.com>
Date: Thu, 3 Jun 2021 19:53:33 +0800
Subject: [PATCH] Update pymilvus version (#5572)

Signed-off-by: Xiangyu Wang <xiangyu.wang@zilliz.com>
---
 tests/python_test/conftest.py           | 2 +-
 tests/python_test/entity/test_insert.py | 2 +-
 tests/python_test/entity/test_search.py | 2 +-
 tests/python_test/requirements.txt      | 2 +-
 tests/python_test/utils.py              | 2 +-
 5 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/tests/python_test/conftest.py b/tests/python_test/conftest.py
index 60de45ccd7..a8ab54f038 100644
--- a/tests/python_test/conftest.py
+++ b/tests/python_test/conftest.py
@@ -3,7 +3,7 @@ import logging
 import socket
 import pytest
 from utils import gen_unique_str
-from milvus import Milvus, DataType
+from pymilvus import Milvus, DataType
 from utils import *
 
 timeout = 60
diff --git a/tests/python_test/entity/test_insert.py b/tests/python_test/entity/test_insert.py
index ac02497551..7b3a1c8a46 100644
--- a/tests/python_test/entity/test_insert.py
+++ b/tests/python_test/entity/test_insert.py
@@ -1,5 +1,5 @@
 import pytest
-from milvus import DataType, ParamError, BaseException
+from pymilvus import DataType, ParamError, BaseException
 from utils import *
 from constants import *
 
diff --git a/tests/python_test/entity/test_search.py b/tests/python_test/entity/test_search.py
index 5117a240af..528e814bb1 100644
--- a/tests/python_test/entity/test_search.py
+++ b/tests/python_test/entity/test_search.py
@@ -6,7 +6,7 @@ from multiprocessing import Pool, Process
 import pytest
 import numpy as np
 
-from milvus import DataType
+from pymilvus import DataType
 from utils import *
 from constants import *
 
diff --git a/tests/python_test/requirements.txt b/tests/python_test/requirements.txt
index 5fec358020..3b7f0ddd7e 100644
--- a/tests/python_test/requirements.txt
+++ b/tests/python_test/requirements.txt
@@ -11,7 +11,7 @@ allure-pytest==2.7.0
 pytest-print==0.2.1
 pytest-level==0.1.1
 pytest-xdist==2.2.1
-pymilvus==2.0a1.dev15
+pymilvus==2.0a1.dev22
 pytest-rerunfailures==9.1.1
 git+https://github.com/Projectplace/pytest-tags
 ndg-httpsclient
diff --git a/tests/python_test/utils.py b/tests/python_test/utils.py
index c1b0134d6e..020199e048 100644
--- a/tests/python_test/utils.py
+++ b/tests/python_test/utils.py
@@ -11,7 +11,7 @@ import time
 import copy
 import numpy as np
 from sklearn import preprocessing
-from milvus import Milvus, DataType
+from pymilvus import Milvus, DataType
 
 port = 19530
 epsilon = 0.000001