Ver Fonte

fix linters

Daniel Gatis há 1 ano atrás
pai
commit
bfafe9e9b4
1 ficheiros alterados com 2 adições e 2 exclusões
  1. 2 2
      rembg/bg.py

+ 2 - 2
rembg/bg.py

@@ -1,6 +1,6 @@
 import io
 from enum import Enum
-from typing import Any, List, Optional, Tuple, Union
+from typing import Any, List, Optional, Tuple, Union, cast
 
 import numpy as np
 import onnxruntime as ort
@@ -192,7 +192,7 @@ def fix_image_orientation(img: PILImage) -> PILImage:
     Returns:
         PILImage: The fixed image.
     """
-    return ImageOps.exif_transpose(img)
+    return cast(PILImage, ImageOps.exif_transpose(img))
 
 
 def download_models() -> None: