Browse Source

fix linter issues

Daniel Gatis 4 months ago
parent
commit
c713191c36
2 changed files with 3 additions and 9 deletions
  1. 2 8
      rembg/bg.py
  2. 1 1
      rembg/session_factory.py

+ 2 - 8
rembg/bg.py

@@ -5,14 +5,8 @@ from typing import Any, List, Optional, Tuple, Union, cast
 
 
 import numpy as np
 import numpy as np
 import onnxruntime as ort
 import onnxruntime as ort
-from cv2 import (
-    BORDER_DEFAULT,
-    MORPH_ELLIPSE,
-    MORPH_OPEN,
-    GaussianBlur,
-    getStructuringElement,
-    morphologyEx,
-)
+from cv2 import (BORDER_DEFAULT, MORPH_ELLIPSE, MORPH_OPEN, GaussianBlur,
+                 getStructuringElement, morphologyEx)
 from PIL import Image, ImageOps
 from PIL import Image, ImageOps
 from PIL.Image import Image as PILImage
 from PIL.Image import Image as PILImage
 from pymatting.alpha.estimate_alpha_cf import estimate_alpha_cf
 from pymatting.alpha.estimate_alpha_cf import estimate_alpha_cf

+ 1 - 1
rembg/session_factory.py

@@ -1,5 +1,5 @@
 import os
 import os
-from typing import Type
+from typing import Optional, Type
 
 
 import onnxruntime as ort
 import onnxruntime as ort