From 4bc064d719d746a64a6eed2f9c7e9562dfe3ba7a Mon Sep 17 00:00:00 2001
From: shiyu22 <shiyu.chen@zilliz.com>
Date: Wed, 2 Nov 2022 19:25:27 +0800
Subject: [PATCH] Add PyOp as base

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
---
 milvus.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/milvus.py b/milvus.py
index 890e530..c464478 100644
--- a/milvus.py
+++ b/milvus.py
@@ -1,11 +1,12 @@
 import numpy as np
 from towhee import register
 from urllib.parse import urlsplit
+from towhee.operator import PyOperator
 from pymilvus import connections, Collection
 
 
 @register(output_schema=['mr'])
-class Milvus:
+class Milvus(PyOperator):
     """
     Milvus ANN index class.
     """