fix linters

Ling137 · Jan 31, 2024 · 8eb37ef · 8eb37ef
1 parent 4beb98d
commit 8eb37ef
Show file tree

Hide file tree

Showing 7 changed files with 35 additions and 21 deletions.
diff --git a/rembg/sessions/dis_anime.py b/rembg/sessions/dis_anime.py
@@ -59,9 +59,11 @@ def download_models(cls, *args, **kwargs):
         fname = f"{cls.name(*args, **kwargs)}.onnx"
         pooch.retrieve(
             "https://github.com/danielgatis/rembg/releases/download/v0.0.0/isnet-anime.onnx",
-            None
-            if cls.checksum_disabled(*args, **kwargs)
-            else "md5:6f184e756bb3bd901c8849220a83e38e",
+            (
+                None
+                if cls.checksum_disabled(*args, **kwargs)
+                else "md5:6f184e756bb3bd901c8849220a83e38e"
+            ),
             fname=fname,
             path=cls.u2net_home(*args, **kwargs),
             progressbar=True,

diff --git a/rembg/sessions/dis_general_use.py b/rembg/sessions/dis_general_use.py
@@ -57,9 +57,11 @@ def download_models(cls, *args, **kwargs):
         fname = f"{cls.name(*args, **kwargs)}.onnx"
         pooch.retrieve(
             "https://github.com/danielgatis/rembg/releases/download/v0.0.0/isnet-general-use.onnx",
-            None
-            if cls.checksum_disabled(*args, **kwargs)
-            else "md5:fc16ebd8b0c10d971d3513d564d01e29",
+            (
+                None
+                if cls.checksum_disabled(*args, **kwargs)
+                else "md5:fc16ebd8b0c10d971d3513d564d01e29"
+            ),
             fname=fname,
             path=cls.u2net_home(*args, **kwargs),
             progressbar=True,

diff --git a/rembg/sessions/silueta.py b/rembg/sessions/silueta.py
@@ -63,9 +63,11 @@ def download_models(cls, *args, **kwargs):
         fname = f"{cls.name()}.onnx"
         pooch.retrieve(
             "https://github.com/danielgatis/rembg/releases/download/v0.0.0/silueta.onnx",
-            None
-            if cls.checksum_disabled(*args, **kwargs)
-            else "md5:55e59e0d8062d2f5d013f4725ee84782",
+            (
+                None
+                if cls.checksum_disabled(*args, **kwargs)
+                else "md5:55e59e0d8062d2f5d013f4725ee84782"
+            ),
             fname=fname,
             path=cls.u2net_home(*args, **kwargs),
             progressbar=True,

diff --git a/rembg/sessions/u2net.py b/rembg/sessions/u2net.py
@@ -61,9 +61,11 @@ def download_models(cls, *args, **kwargs):
         fname = f"{cls.name(*args, **kwargs)}.onnx"
         pooch.retrieve(
             "https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx",
-            None
-            if cls.checksum_disabled(*args, **kwargs)
-            else "md5:60024c5c889badc19c04ad937298a77b",
+            (
+                None
+                if cls.checksum_disabled(*args, **kwargs)
+                else "md5:60024c5c889badc19c04ad937298a77b"
+            ),
             fname=fname,
             path=cls.u2net_home(*args, **kwargs),
             progressbar=True,

diff --git a/rembg/sessions/u2net_cloth_seg.py b/rembg/sessions/u2net_cloth_seg.py
@@ -129,9 +129,11 @@ def download_models(cls, *args, **kwargs):
         fname = f"{cls.name(*args, **kwargs)}.onnx"
         pooch.retrieve(
             "https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_cloth_seg.onnx",
-            None
-            if cls.checksum_disabled(*args, **kwargs)
-            else "md5:2434d1f3cb744e0e49386c906e5a08bb",
+            (
+                None
+                if cls.checksum_disabled(*args, **kwargs)
+                else "md5:2434d1f3cb744e0e49386c906e5a08bb"
+            ),
             fname=fname,
             path=cls.u2net_home(*args, **kwargs),
             progressbar=True,

diff --git a/rembg/sessions/u2net_human_seg.py b/rembg/sessions/u2net_human_seg.py
@@ -61,9 +61,11 @@ def download_models(cls, *args, **kwargs):
         fname = f"{cls.name(*args, **kwargs)}.onnx"
         pooch.retrieve(
             "https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net_human_seg.onnx",
-            None
-            if cls.checksum_disabled(*args, **kwargs)
-            else "md5:c09ddc2e0104f800e3e1bb4652583d1f",
+            (
+                None
+                if cls.checksum_disabled(*args, **kwargs)
+                else "md5:c09ddc2e0104f800e3e1bb4652583d1f"
+            ),
             fname=fname,
             path=cls.u2net_home(*args, **kwargs),
             progressbar=True,

diff --git a/rembg/sessions/u2netp.py b/rembg/sessions/u2netp.py
@@ -53,9 +53,11 @@ def download_models(cls, *args, **kwargs):
         fname = f"{cls.name(*args, **kwargs)}.onnx"
         pooch.retrieve(
             "https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2netp.onnx",
-            None
-            if cls.checksum_disabled(*args, **kwargs)
-            else "md5:8e83ca70e441ab06c318d82300c84806",
+            (
+                None
+                if cls.checksum_disabled(*args, **kwargs)
+                else "md5:8e83ca70e441ab06c318d82300c84806"
+            ),
             fname=fname,
             path=cls.u2net_home(*args, **kwargs),
             progressbar=True,