fix lint

danielgatis · Mar 31, 2023 · 658ab42 · 658ab42
1 parent 54cf4f8
commit 658ab42
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/rembg/cli.py b/rembg/cli.py
@@ -2,7 +2,7 @@
 import sys
 import time
 from enum import Enum
-from typing import IO, Optional, cast
+from typing import IO, Optional, Tuple, cast
 
 import aiohttp
 import click
@@ -377,7 +377,11 @@ def __init__(
             self.ae = ae
             self.om = om
             self.ppm = ppm
-            self.bgc = map(int, bgc.split(",")) if bgc else None
+            self.bgc = (
+                cast(Tuple[int, int, int, int], tuple(map(int, bgc.split(","))))
+                if bgc
+                else None
+            )
 
     class CommonQueryPostParams:
         def __init__(
@@ -413,7 +417,11 @@ def __init__(
             self.ae = ae
             self.om = om
             self.ppm = ppm
-            self.bgc = map(int, bgc.split(",")) if bgc else None
+            self.bgc = (
+                cast(Tuple[int, int, int, int], tuple(map(int, bgc.split(","))))
+                if bgc
+                else None
+            )
 
     def im_without_bg(content: bytes, commons: CommonQueryParams) -> Response:
         return Response(