From 2f46d952540109940584c5047d5fa65b25f51166 Mon Sep 17 00:00:00 2001 From: Xintao Date: Fri, 16 Sep 2022 19:32:42 +0800 Subject: [PATCH] fix pylint --- cog_predict.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/cog_predict.py b/cog_predict.py index dc37cc3..4b16e58 100644 --- a/cog_predict.py +++ b/cog_predict.py @@ -64,13 +64,13 @@ class Predictor(BasePredictor): self.current_version = 'v1.4' def predict( - self, - img: Path = Input(description='Input'), - version: str = Input( - description='GFPGAN version. v1.3: better quality. v1.4: more details and better identity.', - choices=['v1.2', 'v1.3', 'v1.4', 'RestoreFormer'], - default='v1.4'), - scale: float = Input(description='Rescaling factor', default=2), + self, + img: Path = Input(description='Input'), + version: str = Input( + description='GFPGAN version. v1.3: better quality. v1.4: more details and better identity.', + choices=['v1.2', 'v1.3', 'v1.4', 'RestoreFormer'], + default='v1.4'), + scale: float = Input(description='Rescaling factor', default=2), ) -> Path: weight = 0.5 print(img, version, scale, weight)