fix linters

danielgatis · May 23, 2024 · 747ef5e · 747ef5e
1 parent f13083b
commit 747ef5e
Show file tree

Hide file tree

Showing 9 changed files with 9 additions and 9 deletions.
diff --git a/rembg/sessions/base.py b/rembg/sessions/base.py
@@ -46,7 +46,7 @@ def normalize(
         *args,
         **kwargs
     ) -> Dict[str, np.ndarray]:
-        im = img.convert("RGB").resize(size, Image.LANCZOS)
+        im = img.convert("RGB").resize(size, Image.Resampling.LANCZOS)
 
         im_ary = np.array(im)
         im_ary = im_ary / np.max(im_ary)

diff --git a/rembg/sessions/dis_anime.py b/rembg/sessions/dis_anime.py
@@ -40,7 +40,7 @@ def predict(self, img: PILImage, *args, **kwargs) -> List[PILImage]:
         pred = np.squeeze(pred)
 
         mask = Image.fromarray((pred * 255).astype("uint8"), mode="L")
-        mask = mask.resize(img.size, Image.LANCZOS)
+        mask = mask.resize(img.size, Image.Resampling.LANCZOS)
 
         return [mask]
 

diff --git a/rembg/sessions/dis_general_use.py b/rembg/sessions/dis_general_use.py
@@ -36,7 +36,7 @@ def predict(self, img: PILImage, *args, **kwargs) -> List[PILImage]:
         pred = np.squeeze(pred)
 
         mask = Image.fromarray((pred * 255).astype("uint8"), mode="L")
-        mask = mask.resize(img.size, Image.LANCZOS)
+        mask = mask.resize(img.size, Image.Resampling.LANCZOS)
 
         return [mask]
 

diff --git a/rembg/sessions/silueta.py b/rembg/sessions/silueta.py
@@ -42,7 +42,7 @@ def predict(self, img: PILImage, *args, **kwargs) -> List[PILImage]:
         pred = np.squeeze(pred)
 
         mask = Image.fromarray((pred * 255).astype("uint8"), mode="L")
-        mask = mask.resize(img.size, Image.LANCZOS)
+        mask = mask.resize(img.size, Image.Resampling.LANCZOS)
 
         return [mask]
 

diff --git a/rembg/sessions/u2net.py b/rembg/sessions/u2net.py
@@ -42,7 +42,7 @@ def predict(self, img: PILImage, *args, **kwargs) -> List[PILImage]:
         pred = np.squeeze(pred)
 
         mask = Image.fromarray((pred * 255).astype("uint8"), mode="L")
-        mask = mask.resize(img.size, Image.LANCZOS)
+        mask = mask.resize(img.size, Image.Resampling.LANCZOS)
 
         return [mask]
 

diff --git a/rembg/sessions/u2net_cloth_seg.py b/rembg/sessions/u2net_cloth_seg.py
@@ -87,7 +87,7 @@ def predict(self, img: PILImage, *args, **kwargs) -> List[PILImage]:
         pred = np.squeeze(pred, 0)
 
         mask = Image.fromarray(pred.astype("uint8"), mode="L")
-        mask = mask.resize(img.size, Image.LANCZOS)
+        mask = mask.resize(img.size, Image.Resampling.LANCZOS)
 
         masks = []
 

diff --git a/rembg/sessions/u2net_custom.py b/rembg/sessions/u2net_custom.py
@@ -68,7 +68,7 @@ def predict(self, img: PILImage, *args, **kwargs) -> List[PILImage]:
         pred = np.squeeze(pred)
 
         mask = Image.fromarray((pred * 255).astype("uint8"), mode="L")
-        mask = mask.resize(img.size, Image.LANCZOS)
+        mask = mask.resize(img.size, Image.Resampling.LANCZOS)
 
         return [mask]
 

diff --git a/rembg/sessions/u2net_human_seg.py b/rembg/sessions/u2net_human_seg.py
@@ -42,7 +42,7 @@ def predict(self, img: PILImage, *args, **kwargs) -> List[PILImage]:
         pred = np.squeeze(pred)
 
         mask = Image.fromarray((pred * 255).astype("uint8"), mode="L")
-        mask = mask.resize(img.size, Image.LANCZOS)
+        mask = mask.resize(img.size, Image.Resampling.LANCZOS)
 
         return [mask]
 

diff --git a/rembg/sessions/u2netp.py b/rembg/sessions/u2netp.py
@@ -38,7 +38,7 @@ def predict(self, img: PILImage, *args, **kwargs) -> List[PILImage]:
         pred = np.squeeze(pred)
 
         mask = Image.fromarray((pred * 255).astype("uint8"), mode="L")
-        mask = mask.resize(img.size, Image.LANCZOS)
+        mask = mask.resize(img.size, Image.Resampling.LANCZOS)
 
         return [mask]