From 29eabd19247cd9fd03b9f9e10220ca411cf2974d Mon Sep 17 00:00:00 2001
From: "junjie.jiang" <junjie.jiang@zilliz.com>
Date: Fri, 10 Feb 2023 11:54:27 +0800
Subject: [PATCH] rm _types

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
---
 mpvit.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mpvit.py b/mpvit.py
index 4058b4d..790fc04 100644
--- a/mpvit.py
+++ b/mpvit.py
@@ -54,7 +54,7 @@ class MPViT(NNOperator):
         self.skip_tfms = skip_preprocess
         self.device = device
 
-    def __call__(self, data: Union[List[towhee._types.Image], towhee._types.Image]):
+    def __call__(self, data: Union[List['towhee.types.Image'], 'towhee.types.Image']):
         if not isinstance(data, list):
             imgs = [data]
         else:
@@ -77,6 +77,6 @@ class MPViT(NNOperator):
         return vecs
 
     @arg(1, to_image_color('RGB'))
-    def convert_img(self, img: towhee._types.Image):
+    def convert_img(self, img: 'towhee.types.Image'):
         img = PILImage.fromarray(img.astype('uint8'), 'RGB')
         return img