chore: install milvus

Signed-off-by: jupyterjazz <[email protected]>
docarray · JoanFM · Jul 19, 2023 · Jul 2, 2023 · Jul 2, 2023 · Jul 13, 2023
commit 06d06022b023a52f39645f3656f97fc017f4682a
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('pymilvus', raise_error=True)
+        import_library('milvus', 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,6 +43,7 @@
     'boto3': '"docarray[aws]"',
     'botocore': '"docarray[aws]"',
     'redis': '"docarray[redis]"',
+    'milvus': '"docarray[milvus]"'
 }
 
 

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -76,7 +76,7 @@ torch = ["torch"]
 web = ["fastapi"]
 qdrant = ["qdrant-client"]
 weaviate = ["weaviate-client"]
-pymilvus = ["pymilvus"]
+milvus = ["pymilvus"]
 redis = ['redis']
 
 # all