From 853effa72fe3ccb6fe0e55809f4953281a439488 Mon Sep 17 00:00:00 2001 From: "junjie.jiang" Date: Mon, 30 Jan 2023 11:33:07 +0800 Subject: [PATCH] rm image dep Signed-off-by: junjie.jiang --- isc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/isc.py b/isc.py index f70f5c4..8a5ee07 100644 --- a/isc.py +++ b/isc.py @@ -86,8 +86,8 @@ class Isc(NNOperator): std=self.backbone.default_cfg['std']) ]) - def __call__(self, data: Union[List[towhee._types.Image], towhee._types.Image]): - if isinstance(data, towhee._types.Image): + def __call__(self, data: Union[List['towhee.types.Image'], 'towhee.types.Image']): + if not isinstance(data, list): imgs = [data] else: imgs = data