rm _types

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
1 year ago · 86b69319fc
1 changed files with 1 additions and 1 deletions
--- a/timm_image.py
+++ b/timm_image.py
@ -139,7 +139,7 @@ class TimmImage(NNOperator):
        return config

    @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