fix: installing pymilvus

Signed-off-by: jupyterjazz <saba.sturua@jina.ai>
JoanFM · Jul 19, 2023 · Jul 2, 2023 · Jul 2, 2023 · Jul 13, 2023 · Jul 13, 2023
commit 30d133b050fbab2a5ff98d8f65c41e0a3b287aae
diff --git a/docarray/index/__init__.py b/docarray/index/__init__.py
@@ -37,7 +37,7 @@ def __getattr__(name: str):
         import_library('weaviate', raise_error=True)
         import docarray.index.backends.weaviate as lib
     elif name == 'MilvusDocumentIndex':
-        import_library('milvus', raise_error=True)
+        import_library('pymilvus', raise_error=True)
         import docarray.index.backends.milvus as lib
     elif name == 'RedisDocumentIndex':
         import_library('redis', raise_error=True)

diff --git a/docarray/utils/_internal/misc.py b/docarray/utils/_internal/misc.py
@@ -43,7 +43,7 @@
     'boto3': '"docarray[aws]"',
     'botocore': '"docarray[aws]"',
     'redis': '"docarray[redis]"',
-    'milvus': '"docarray[milvus]"'
+    'pymilvus': '"docarray[milvus]"',
 }