diff --git a/milvus_client.py b/milvus_client.py index 0647991..e8ac4a2 100644 --- a/milvus_client.py +++ b/milvus_client.py @@ -1,6 +1,6 @@ import uuid import logging -from towhee.operator import PyOperator +from towhee.operator import PyOperator, SharedType from pymilvus import connections, Collection @@ -39,3 +39,7 @@ class MilvusClient(PyOperator): if mr.err_count > 0: raise RuntimeError("Insert to milvus failed") return None + + @property + def shared_type(self): + return SharedType.Shareable