fix: fix mypy errors

Author: boomb0om

DPF/filters/videos/image_filter_adapter.py

@@ -69,7 +69,7 @@ def preprocess_data(
         frame = iio.imread(io.BytesIO(video_bytes), index=frame_index, plugin="pyav")
 
         buff = io.BytesIO()
-        Image.fromarray(frame).convert('RGB').save(buff, format='JPEG', quality=95)
+        Image.fromarray(frame).convert('RGB').save(buff, format='JPEG', quality=95)  # type: ignore
         modality2data['image'] = buff.getvalue()
         metadata[self.image_filter.key_column] = ''
         return key, self.image_filter.preprocess_data(modality2data, metadata)

DPF/filters/videos/pllava_filter.py

@@ -31,7 +31,7 @@ def load_video(video_bytes: BytesIO, num_segments: int = 8, return_msg: bool = F
     frame_indices = get_index(num_frames, num_segments)
     images_group = []
     for frame_index in frame_indices:
-        img = Image.fromarray(vr[frame_index].asnumpy())
+        img = Image.fromarray(vr[frame_index].asnumpy())  # type: ignore
         images_group.append(transforms(img))
     if return_msg:
         fps = float(vr.get_avg_fps())

DPF/transforms/image_resize_transforms.py

@@ -41,7 +41,7 @@ def _process_filepath(self, data: TransformsFileData) -> TransformsFileData:
         width, height = self.resizer.get_new_size(img.width, img.height)
 
         if (width, height) != (img.width, img.height):
-            img = img.resize((width, height))  # type: ignore
+            img = img.resize((width, height))
             img.save(filepath, format=self.img_format)
 
         return TransformsFileData(filepath, {'width': width, 'height': height})

DPF/utils/image_utils.py

@@ -7,17 +7,17 @@ def read_image_rgb(path: str, force_rgb: bool = True) -> Image.Image:
     pil_img = Image.open(path)
     pil_img.load()  # type: ignore
     if pil_img.format == "PNG" and pil_img.mode != "RGBA":
-        pil_img = pil_img.convert("RGBA")  # type: ignore
+        pil_img = pil_img.convert("RGBA")
     if force_rgb:
-        pil_img = pil_img.convert("RGB")  # type: ignore
+        pil_img = pil_img.convert("RGB")
     return pil_img
 
 
 def read_image_rgb_from_bytes(img_bytes: bytes, force_rgb: bool = True) -> Image.Image:
     pil_img = Image.open(BytesIO(img_bytes))
     pil_img.load()  # type: ignore
     if pil_img.format == "PNG" and pil_img.mode != "RGBA":
-        pil_img = pil_img.convert("RGBA")  # type: ignore
+        pil_img = pil_img.convert("RGBA")
     if force_rgb:
-        pil_img = pil_img.convert("RGB")   # type: ignore
+        pil_img = pil_img.convert("RGB")
     return pil_img