From 552b05d80761d762ac3b86169eee39b5ba5d2d9e Mon Sep 17 00:00:00 2001
From: "junjie.jiang" <junjie.jiang@zilliz.com>
Date: Fri, 10 Feb 2023 11:58:52 +0800
Subject: [PATCH] rm _types

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
---
 swag.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/swag.py b/swag.py
index 7bbc412..cfaed1c 100644
--- a/swag.py
+++ b/swag.py
@@ -59,7 +59,7 @@ class Swag(NNOperator):
         self.model.eval()
 
     @arg(1, to_image_color('RGB'))
-    def __call__(self, img: towhee._types.Image) -> numpy.ndarray:
+    def __call__(self, img: 'towhee.types.Image') -> numpy.ndarray:
         img = PILImage.fromarray(img.astype('uint8'), 'RGB')
         if not self.skip_tfms:
             img = self.tfms(img).unsqueeze(0)