apply lint

ssube · Feb 14, 2023 · 1289c56 · 1289c56
1 parent cc7cafa
commit 1289c56
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 8 deletions.
diff --git a/api/onnx_web/chain/correct_gfpgan.py b/api/onnx_web/chain/correct_gfpgan.py
@@ -4,7 +4,6 @@
 import numpy as np
 from PIL import Image
 
-
 from ..params import DeviceParams, ImageParams, StageParams, UpscaleParams
 from ..server.device_pool import JobContext
 from ..utils import ServerContext, run_gc

diff --git a/api/onnx_web/chain/upscale_resrgan.py b/api/onnx_web/chain/upscale_resrgan.py
@@ -76,9 +76,7 @@ def load_resrgan(
     logger.debug("loading Real ESRGAN upscale model from %s", model_path)
 
     # TODO: shouldn't need the PTH file
-    model_path_pth = path.join(
-        server.cache_path, ("%s.pth" % params.upscale_model)
-    )
+    model_path_pth = path.join(server.cache_path, ("%s.pth" % params.upscale_model))
     upsampler = RealESRGANer(
         scale=params.scale,
         model_path=model_path_pth,

diff --git a/api/onnx_web/diffusion/load.py b/api/onnx_web/diffusion/load.py
@@ -20,7 +20,7 @@
 
 try:
     from diffusers import DEISMultistepScheduler
-except:
+except ImportError:
     from .stub_scheduler import StubScheduler as DEISMultistepScheduler
 
 from ..params import DeviceParams, Size

diff --git a/api/onnx_web/diffusion/stub_scheduler.py b/api/onnx_web/diffusion/stub_scheduler.py
@@ -1,6 +1,8 @@
 from typing import Any
 
 
-class StubScheduler():
-  def step(self, model_output: Any, timestep: int, sample: Any, return_dict: bool = True) -> None:
-    raise NotImplemented("scheduler not available, try updating diffusers")
+class StubScheduler:
+    def step(
+        self, model_output: Any, timestep: int, sample: Any, return_dict: bool = True
+    ) -> None:
+        raise NotImplementedError("scheduler not available, try updating diffusers")