Enable Pyrefly in fbcode/aml/augly

Summary:
Automated migration to enable Pyrefly type checking for `fbcode/aml/augly`.

- Added `python.set_pyrefly(True)` to PACKAGE file
- Suppressed pre-existing type errors

Pyrefly is Meta's next-generation Python type checker, replacing Pyre.

If you encounter issues, you can revert the PACKAGE change by removing
the `python.set_pyrefly(True)` line.
#pyreupgrade

Differential Revision: D107126783

fbshipit-source-id: cee788e6240bffe2034211c2d2cb182877429073

Author: generatedunixname89002005307016

augly/audio/functional.py

@@ -104,15 +104,20 @@ def add_background_noise(
 
     background_audio *= desired_bg_rms / bg_rms
 
+    # pyrefly: ignore [missing-attribute]
     while background_audio.shape[-1] < audio.shape[-1]:
+        # pyrefly: ignore [missing-attribute]
         axis = 0 if background_audio.ndim == 1 else 1
         background_audio = np.concatenate(
             (background_audio, background_audio), axis=axis
         )
 
     background_audio = (
+        # pyrefly: ignore [unsupported-operation]
         background_audio[: audio.shape[-1]]
+        # pyrefly: ignore [missing-attribute]
         if background_audio.ndim == 1
+        # pyrefly: ignore [bad-index, unsupported-operation]
         else background_audio[:, : audio.shape[-1]]
     )
 

augly/audio/utils.py

@@ -86,8 +86,10 @@ def check_random_state(seed: RNGSeed | None) -> RNG:
         Otherwise raise ValueError.
     """
     if seed is None or seed is np.random:
+        # pyrefly: ignore [missing-attribute]
         return np.random.mtrand._rand
     if isinstance(seed, numbers.Integral):
+        # pyrefly: ignore [bad-argument-type]
         return np.random.RandomState(cast(int, seed))
     if isinstance(seed, (np.random.RandomState, np.random.Generator)):
         return seed

augly/image/functional.py

@@ -1842,6 +1842,7 @@ def overlay_random_text_with_background(
     random.seed(seed)
 
     if font_file is None:
+        # pyrefly: ignore [bad-assignment]
         font_file = utils.SOCIAL_MEDIA_OVERLAY_FONTS
 
     font_files = font_file if isinstance(font_file, list) else [font_file]
@@ -1877,6 +1878,7 @@ def overlay_random_text_with_background(
 
     for overlay_idx in range(num_overlays):
         cur_font_file = random.choice(font_files)
+        # pyrefly: ignore [bad-argument-type]
         local_font_path = utils.pathmgr.get_local_path(cur_font_file)
 
         text = random.choice(phrases)
@@ -2054,6 +2056,7 @@ def overlay_random_text_with_background(
 def overlay_text(
     image: str | Image.Image,
     output_path: str | None = None,
+    # pyrefly: ignore [bad-function-definition]
     text: list[int | list[int]] = utils.DEFAULT_TEXT_INDICES,
     font_file: str = utils.FONT_PATH,
     font_size: float = 0.15,
@@ -2472,6 +2475,7 @@ def perspective_transform(
     func_kwargs = imutils.get_func_kwargs(metadata, locals())
     src_mode = image.mode
 
+    # pyrefly: ignore [bad-argument-type]
     rng = np.random.RandomState(seed) if seed is not None else np.random
     width, height = image.size
 
@@ -2616,6 +2620,7 @@ def random_noise(
     src_mode = image.mode
 
     if seed is not None:
+        # pyrefly: ignore [bad-argument-type]
         np.random.seed(seed=seed)
 
     np_image = np.asarray(image).astype(np.float32)
@@ -3158,6 +3163,7 @@ def shuffle_pixels(
 
     @returns: the augmented PIL Image
     """
+    # pyrefly: ignore [bad-argument-type]
     np.random.seed(seed)
 
     image = imutils.validate_and_load_image(image)
@@ -3307,6 +3313,7 @@ def split_and_shuffle(
 
     @returns: the augmented PIL Image
     """
+    # pyrefly: ignore [bad-argument-type]
     np.random.seed(seed)
 
     image = imutils.validate_and_load_image(image)

augly/image/transforms.py

@@ -1416,6 +1416,7 @@ def apply_transform(
 class OverlayText(BaseTransform):
     def __init__(
         self,
+        # pyrefly: ignore [bad-function-definition]
         text: list[int | list[int]] = utils.DEFAULT_TEXT_INDICES,
         font_file: str = utils.FONT_PATH,
         font_size: float = 0.15,
@@ -2643,6 +2644,7 @@ def apply_random_transform(
         """
         return F.change_aspect_ratio(
             image,
+            # pyrefly: ignore [bad-argument-type]
             ratio=self.chosen_value,
             metadata=metadata,
             bboxes=bboxes,
@@ -2693,6 +2695,7 @@ def apply_random_transform(
         """
         return F.blur(
             image,
+            # pyrefly: ignore [bad-argument-type]
             radius=self.chosen_value,
             metadata=metadata,
             bboxes=bboxes,
@@ -2743,6 +2746,7 @@ def apply_random_transform(
         """
         return F.brightness(
             image,
+            # pyrefly: ignore [bad-argument-type]
             factor=self.chosen_value,
             metadata=metadata,
             bboxes=bboxes,
@@ -2905,6 +2909,7 @@ def apply_random_transform(
         """
         return F.pixelization(
             image,
+            # pyrefly: ignore [bad-argument-type]
             ratio=self.chosen_value,
             metadata=metadata,
             bboxes=bboxes,
@@ -2969,6 +2974,7 @@ def apply_random_transform(
         """
         return F.rotate(
             image,
+            # pyrefly: ignore [bad-argument-type]
             degrees=self.chosen_value,
             expand=self.expand,
             fill_color=self.fill_color,

augly/text/augmenters/case.py

@@ -20,7 +20,10 @@
 class CaseChanger:
     def __init__(self, case: str, seed: int | None) -> None:
         self.rng: np.random.RandomState | types.ModuleType = (
-            np.random.RandomState(seed) if seed is not None else np.random
+            # pyrefly: ignore [bad-argument-type]
+            np.random.RandomState(seed)
+            if seed is not None
+            else np.random
         )
         self.case = case
 

augly/text/augmenters/contraction.py

@@ -53,6 +53,7 @@ def __init__(
         self.aug_p = aug_p
         self.contraction_mapping = self.get_mapping(mapping)
         self.max_contraction_length = max_contraction_length
+        # pyrefly: ignore [bad-argument-type, bad-assignment]
         self.rng = np.random.RandomState(seed) if seed is not None else np.random
 
     def get_mapping(self, mapping: str | dict[str, Any] | None) -> ContractionMapping:

augly/text/augmenters/word_replacement.py

@@ -84,6 +84,7 @@ def substitute(self, data: str) -> str:
         filtered_word_idxes = self.pre_skip_aug(tokens)
 
         if self.priority_words is None:
+            # pyrefly: ignore [bad-assignment]
             self.priority_words = self.word_mapping.mapping.keys()
 
         aug_word_idxes = set(

augly/video/augmenters/cv2/text.py

@@ -104,6 +104,7 @@ def random_fonts(
                     chars = [chr(c) for c in pickle.load(f)]
             else:
                 chars = list(string.ascii_letters + string.punctuation)
+            # pyrefly: ignore [invalid-yield]
             yield font, chars
 
     def random_fontscales(

augly/video/functional.py

@@ -821,6 +821,7 @@ def insert_in_background(
     video_duration = float(video_info["duration"])
     width, height = video_info["width"], video_info["height"]
 
+    # pyrefly: ignore [bad-argument-type]
     rng = np.random.RandomState(seed) if seed is not None else np.random
 
     video_paths = []
@@ -955,6 +956,7 @@ def insert_in_background_multiple(
             src_ids
         ), "src_ids need to be specified for the main video and all additional videos."
     func_kwargs = helpers.get_func_kwargs(metadata, locals(), video_path)
+    # pyrefly: ignore [bad-argument-type]
     rng = np.random.RandomState(seed) if seed is not None else np.random
 
     local_path = utils.pathmgr.get_local_path(video_path)
@@ -1840,6 +1842,7 @@ def perspective_transform_and_shake(
     )
 
     duration = float(helpers.get_video_info(video_path)["duration"])
+    # pyrefly: ignore [bad-argument-type]
     rng = np.random.RandomState(seed) if seed is not None else np.random
 
     def get_dx_dy(frame_number: int) -> dict:
@@ -1989,6 +1992,7 @@ def replace_with_color_frames(
         if output_path != video_path:
             shutil.copy(video_path, output_path)
         if metadata is not None:
+            # pyrefly: ignore [bad-argument-type]
             helpers.get_metadata(metadata=metadata, **func_kwargs)
         return output_path or video_path
 
@@ -2009,6 +2013,7 @@ def replace_with_color_frames(
         if offset_factor == 0 and duration_factor == 1.0:
             shutil.copy(color_path, output_path)
             if metadata is not None:
+                # pyrefly: ignore [bad-argument-type]
                 helpers.get_metadata(metadata=metadata, **func_kwargs)
             return output_path or video_path
 
@@ -2032,6 +2037,7 @@ def replace_with_color_frames(
         )
 
     if metadata is not None:
+        # pyrefly: ignore [bad-argument-type]
         helpers.get_metadata(metadata=metadata, **func_kwargs)
 
     return output_path or video_path

augly/video/transforms.py

@@ -1951,7 +1951,14 @@ def apply_transform(
         @returns: the path to the augmented video
         """
         return F.resize(
-            video_path, output_path, self.height, self.width, metadata=metadata
+            # pyrefly: ignore [bad-argument-type]
+            video_path,
+            output_path,
+            # pyrefly: ignore [bad-argument-type]
+            self.height,
+            # pyrefly: ignore [bad-argument-type]
+            self.width,
+            metadata=metadata,
         )
 
 
@@ -2345,7 +2352,12 @@ def apply_random_transform(
         @returns: the path to the augmented video
         """
         return F.change_aspect_ratio(
-            video_path, output_path, ratio=self.chosen_value, metadata=metadata
+            # pyrefly: ignore [bad-argument-type]
+            video_path,
+            output_path,
+            # pyrefly: ignore [bad-argument-type]
+            ratio=self.chosen_value,
+            metadata=metadata,
         )
 
 
@@ -2385,7 +2397,12 @@ def apply_random_transform(
         @returns: the path to the augmented video
         """
         return F.blur(
-            video_path, output_path, sigma=self.chosen_value, metadata=metadata
+            # pyrefly: ignore [bad-argument-type]
+            video_path,
+            output_path,
+            # pyrefly: ignore [bad-argument-type]
+            sigma=self.chosen_value,
+            metadata=metadata,
         )
 
 
@@ -2427,6 +2444,7 @@ def apply_random_transform(
         return F.brightness(
             video_path,
             output_path,
+            # pyrefly: ignore [bad-argument-type]
             level=self.chosen_value,
             metadata=metadata,
         )
@@ -2468,7 +2486,12 @@ def apply_random_transform(
         @returns: the path to the augmented video
         """
         return F.contrast(
-            video_path, output_path, level=self.chosen_value, metadata=metadata
+            # pyrefly: ignore [bad-argument-type]
+            video_path,
+            output_path,
+            # pyrefly: ignore [bad-argument-type]
+            level=self.chosen_value,
+            metadata=metadata,
         )
 
 
@@ -2577,6 +2600,7 @@ def apply_random_transform(
         return F.encoding_quality(
             video_path,
             output_path,
+            # pyrefly: ignore [bad-argument-type]
             quality=int(self.chosen_value),
             metadata=metadata,
         )
@@ -2617,6 +2641,7 @@ def apply_random_transform(
 
         @returns: the path to the augmented video
         """
+        # pyrefly: ignore [bad-argument-type]
         return F.fps(video_path, output_path, fps=self.chosen_value, metadata=metadata)
 
 
@@ -2654,7 +2679,12 @@ def apply_random_transform(
         @returns: the path to the augmented video
         """
         return F.add_noise(
-            video_path, output_path, level=int(self.chosen_value), metadata=metadata
+            # pyrefly: ignore [bad-argument-type]
+            video_path,
+            output_path,
+            # pyrefly: ignore [bad-argument-type]
+            level=int(self.chosen_value),
+            metadata=metadata,
         )
 
 
@@ -2698,6 +2728,7 @@ def apply_random_transform(
         return F.pixelization(
             video_path,
             output_path,
+            # pyrefly: ignore [bad-argument-type]
             ratio=self.chosen_value,
             metadata=metadata,
         )
@@ -2737,7 +2768,12 @@ def apply_random_transform(
         @returns: the path to the augmented video
         """
         return F.rotate(
-            video_path, output_path, degrees=self.chosen_value, metadata=metadata
+            # pyrefly: ignore [bad-argument-type]
+            video_path,
+            output_path,
+            # pyrefly: ignore [bad-argument-type]
+            degrees=self.chosen_value,
+            metadata=metadata,
         )
 
 
@@ -2781,5 +2817,10 @@ def apply_random_transform(
         @returns: the path to the augmented video
         """
         return F.change_video_speed(
-            video_path, output_path, factor=self.chosen_value, metadata=metadata
+            # pyrefly: ignore [bad-argument-type]
+            video_path,
+            output_path,
+            # pyrefly: ignore [bad-argument-type]
+            factor=self.chosen_value,
+            metadata=metadata,
         )