update

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
2 years ago · a1dc5133f8
1 changed files with 7 additions and 0 deletions
--- a/milvus_client.py
+++ b/milvus_client.py
@ -43,3 +43,10 @@ class MilvusClient(PyOperator):
    @property
    def shared_type(self):
        return SharedType.Shareable    
+
+    def __del__(self):
+        if connections.has_connection(self._connect_name):
+            try:
+                connections.disconnect(self._connect_name)
+            except:
+                pass