From 658ab422591c341b3cb178ecae854210c0b2d000 Mon Sep 17 00:00:00 2001 From: Daniel Gatis Date: Fri, 31 Mar 2023 13:01:17 -0300 Subject: [PATCH] fix lint --- rembg/cli.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/rembg/cli.py b/rembg/cli.py index a69a671..cfb1760 100644 --- a/rembg/cli.py +++ b/rembg/cli.py @@ -2,7 +2,7 @@ import pathlib 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 s(port: int, log_level: str, threads: int) -> None: 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 s(port: int, log_level: str, threads: int) -> None: 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(